【仙女踩坑实录】python3.6 anaconda安装pytorch/python/torchvision/cudatoolkit以及版本对应

 

图自https://blog.csdn.net/abc13526222160/article/details/86497684

然而，经常会遇到某个源找不到某个包的情况，于是，在anaconda3中，可以直接手动按照如下方式配置（python版本为3.6）。

首先，用pip安装pytorch，版本为1.4.0，不知道为什么用清华源下载贼慢，换成豆瓣好很多，网速不好可以设置超时。

pip install --upgrade torch==1.4.0 -i https://pypi.douban.com/simple --default-timeout=10000

然后，用pip安装torchvision，版本为0.5.0，这步很快。

pip install torchvision==0.5.0 -i https://pypi.douban.com/simple

在这之后，用conda安装cudatoolkit，版本为10.1.243，用的中科大的源。

conda install cudatoolkit=10.1.243 -c https://mirrors.ustc.edu.cn/anaconda/pkgs/main

最后，用conda安装cudnn，版本为6.0。

conda install cudnn==6.0 -c https://mirrors.ustc.edu.cn/anaconda/pkgs/free

至此，安装成功。

注：这只是我安装成功的一种方法，可能与网速和人品都有关系，并不代表其他方法不可以，仅供参考。

测试一下。

#判断是否安装了cuda
import torch
print(torch.cuda.is_available())  #返回True则说明已经安装了cuda
#判断是否安装了cuDNN
from torch.backends import  cudnn 
print(cudnn.is_available())  #返回True则说明已经安装了cuDNN

今天又是从坑里跳出来的一天，开心！