解决通过anaconda安装pytorch时候,出现的cuda版本官网没有的情况
在学习深度学习框架pytorch的时候需要进行安装,由于之前使用过tensorflow的gpu版本,所以安装过cuda10.0版本,因此在配置pytorch的时候就不需要重新安装cuda(因为下起来好麻烦),但是在官网上准备安装的时候发现问题,并没有cuda10.0这个cuda版本的选择,弄了半天才解决掉,所以想在这里将遇到的问题解决方法分享出来,希望遇到同样问题的朋友可以试试看能否解决,下面就是安装过程(除cuda10.0)以及安装过程中遇到的问题。
通过anaconda创建pytorch环境
- 首先通过anaconda这个管理工具创建一个专门用于pytorch的环境,打开控制台窗口。输入:
conda create --name env_pytorch python=3.7
其中env_pytorch
为环境名称,自己可以更改
python版本也可以根据自己的需求更换。
- 然后通过查看已安装环境,看环境有没有安装成功。
conda env list
- 接下来切换至创建好的环境进行安装:
activate env_pytorch
如图可见,括号里的名称为你当前的环境,并且列表中带有 * 的环境为当前环境。
问题1:安装pytorch没有cuda10.0版本
进入官网: https://pytorch.org/get-started/locally/.
根据自己的需求选择相应的选择,我在安装的时候最大的问题就是下面并没有cuda10.0的版本,那么我们直接复制10.1下面的安装命令,并且将它改为10.0
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
然后回车,之后会询问你 y/n ?
,选择 y
即可同意安装。
但是问题2就出现了…
问题2:安装进度条不动
会发现在安装pytorch的时候进度条几乎没有在动,一度让人怀疑挂掉,别急,将下载源头更换为清华源即可(会快很多哦),如果不知道怎么更换为清华源,别急,参考以下方法即可(应该不会有问题):
- 先查看自己有没有已经安装镜像源,在控制台输入命令:
conda config --show
看有没有显示已经安装的镜像源,如下所示:
如果没有也没问题,执行下面的命令安装pytorch的清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后再执行以下命令进行安装:
conda install pytorch torchvision cudatoolkit=10.0
- 注意:这个命令和之前的还是有区别的,细心的同学有没有发现呀,后面的
-c pytorch
被删掉了,这样就可以了。
接下来就可以体验到你家正常网速的下载速度了。
。。。
嗷对,还有一点,这样的配置下安装pytorch为了匹配 cuda10.0
而安装的版本为pytorch 1.2.0
。
测试
等安装完成之后,用控制台打开安装了pytorch的python环境(一定要在安装好的环境下打开哦):
python
进入python环境后,执行以下代码测试查看安装版本以及是否可以用gpu加速:
import torch
print(torch.__version__) # 查看pytorch版本
print('gpu:',torch.cuda.is_available()) # gpu加速是否可用
输出如下结果,那么就大功告成啦!
end…