Pytorch安装速度过慢解决办法

写在最前面:
在虚拟环境中(anaconda的虚拟环境)利用Anaconda中的conda命令安装pytorch时,不需要安装CUDA和对应的cudnn,pytorch官网给出的安装命令会自动安装适合与本地环境的CUDA以及cudnn。
安装过程如下:

1、更换Anconda的下载源为国内的下载源,具体方法见博客:

更换anaconda下载源为国内源

2、到pytorch官网寻找对应的安装命令

在这里,本机选用安装CUDA版本为10.1所对应的安装命令

Pytorch安装速度过慢解决办法_第1张图片

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

注意!!!坑出现了!!!,官网给出的安装命令中,-C参数指明了Pytorch安装的下载源,在这里执行该命令的时候,一定要把-C参数连带后面的pytorch去掉,否则龟速会再次降临!

3、安装完成,在对应的虚拟环境中查看该环境中的安装包以检测pytorch是否安装成功。

在安装的虚拟环境中通过conda list命令查看本环境的安装包:
Pytorch安装速度过慢解决办法_第2张图片查询的结果如下:
Pytorch安装速度过慢解决办法_第3张图片
Pytorch安装速度过慢解决办法_第4张图片
可以看到,本机的pytoch安装成功!

你可能感兴趣的:(计算机视觉实践)