在学习深度学习框架pytorch的时候需要进行安装,由于之前使用过tensorflow的gpu版本,所以安装过cuda10.0版本,因此在配置pytorch的时候就不需要重新安装cuda(因为下起来好麻烦),但是在官网上准备安装的时候发现问题,并没有cuda10.0这个cuda版本的选择,弄了半天才解决掉,所以想在这里将遇到的问题解决方法分享出来,希望遇到同样问题的朋友可以试试看能否解决,下面就是安装过程(除cuda10.0)以及安装过程中遇到的问题。
其中env_pytorch为环境名称,自己可以更改 python版本也可以根据自己的需求更换。
然后通过查看已安装环境,看环境有没有安装成功。 conda env list 接下来切换至创建好的环境进行安装: activate env_pytorch如图可见,括号里的名称为你当前的环境,并且列表中带有 * 的环境为当前环境。
进入官网: 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就出现了…
会发现在安装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…
