安装环境:
cuda: 8.0.61 + python 3.6.5 + Linux
安装步骤:
1:后台新建安装环境
创建新的虚拟环境:conda create -n pytorch python=python3.6(这个自己选择,也就是Anoconda的好处)
2: 进入虚拟环境搭建pytorch环境:source activate pytorch
3: 下载安装:(需要根据自己的环境改变安装命令)
conda install pytorch torchvision -c pytorch
如果你的环境跟我不同 可以参考下篇博客:https://blog.csdn.net/qq_41149269/article/details/81267591
4:测试安装成功:在后台输入python--------import pytorch
如果无报错为成功
参考博客:
https://blog.csdn.net/sinat_34611224/article/details/83026924