第一步:安装cuda8和cudnn。
这位博主的文章讲的很清楚,这里就不在赘述了:
cuda8.0 cuda 8.0 安装pytorch_qq_42331243的博客-CSDN博客_cuda8.0 pytorch
第二步:安装pytorch。
我这里的情况是安装pytorch时出现问题,离线安装和anaconda安装都不成功,由于水平有限,估计也解决不了,最后我这里选择了miniconda安装的方式。
Miniconda — conda documentation
miniconda默认源网速很慢,这里需要切换到清华源。
配置文件地址:
配置内容:
channels:
- defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
配置完成之后就可以安装pytorch了:
Previous PyTorch Versions | PyTorch
这里给出安装命令:
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
第三步:验证安装结果
如果能输出这个结果,应该就表示安装成功了。