上周装了一下tensorflow-GPU版本的,今天来装一下pytorch-GPU版本,有了之前的tensorflow的安装,装pytorch就相对容易了,以下是我边装边写的,尽量简单易懂。
本文在介绍pytorch前默认安装了Anaconda,cuda,cudnn
- 没装的话可以看我之前的tensorflow的博客
同时需要清楚自己的cuda版本,cudnn版本
我的是:cuda9.0,cudnn7.0.5 系统Win10
cuda版本查看方法: 命令行输入 nvcc -V
-
因为2019年4月16日和25日清华和中科大分别宣布停止Anaconda镜像服务,这之后装大概率会出现如下错误(网外网速慢):
-
就按下面三条命令来,下载还是很快
# 第一句
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# 第二句
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 第三句
conda config --set show_channel_urls yes
# 第四句
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- 然后再输安装命令即可,速度真的超级超级超级快的!!!!!
# 9.0是版本号,如果和我不一样的话,记得换一下(-c pytorch这个后缀不要加)
conda install pytorch torchvision cudatoolkit=9.0
- 安装成功测试代码
import torch
torch.__version__ # 返回版本号
torch.cuda.is_available() # 返回True