分享安装pytorch过程以及遇到的一些小问题

解决通过anaconda安装pytorch时候,出现的cuda版本官网没有的情况

在学习深度学习框架pytorch的时候需要进行安装,由于之前使用过tensorflow的gpu版本,所以安装过cuda10.0版本,因此在配置pytorch的时候就不需要重新安装cuda(因为下起来好麻烦),但是在官网上准备安装的时候发现问题,并没有cuda10.0这个cuda版本的选择,弄了半天才解决掉,所以想在这里将遇到的问题解决方法分享出来,希望遇到同样问题的朋友可以试试看能否解决,下面就是安装过程(除cuda10.0)以及安装过程中遇到的问题。

通过anaconda创建pytorch环境

  1. 首先通过anaconda这个管理工具创建一个专门用于pytorch的环境,打开控制台窗口。输入:
conda create --name env_pytorch python=3.7

其中env_pytorch为环境名称,自己可以更改
python版本也可以根据自己的需求更换。

  1. 然后通过查看已安装环境,看环境有没有安装成功。
conda env list
  1. 接下来切换至创建好的环境进行安装:
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的时候进度条几乎没有在动,一度让人怀疑挂掉,别急,将下载源头更换为清华源即可(会快很多哦),如果不知道怎么更换为清华源,别急,参考以下方法即可(应该不会有问题):

  1. 先查看自己有没有已经安装镜像源,在控制台输入命令:
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…

  • 6
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值