写在最前面:
在虚拟环境中(anaconda的虚拟环境)利用Anaconda中的conda命令安装pytorch时,不需要
安装CUDA和对应的cudnn,pytorch官网给出的安装命令会自动安装适合与本地环境的CUDA以及cudnn。
安装过程如下:
1、更换Anconda的下载源为国内的下载源,具体方法见博客:
2、到pytorch官网寻找对应的安装命令
在这里,本机选用安装CUDA版本为10.1所对应的安装命令
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
注意!!!
坑出现了!!!,官网给出的安装命令中,-C参数指明了Pytorch安装的下载源,在这里执行该命令的时候,一定要把-C参数连带后面的pytorch去掉
,否则龟速会再次降临!
3、安装完成,在对应的虚拟环境中查看该环境中的安装包以检测pytorch是否安装成功。
在安装的虚拟环境中通过conda list
命令查看本环境的安装包:
查询的结果如下:
可以看到,本机的pytoch安装成功!