anaconda环境安装GPU版的pytorch

进入Anaconda Promt,在这里面创建虚拟环境
在这里插入图片描述

找到Anaconda目录中的envs,记住这个的路径在这里插入图片描述
一路转到这个envs目录下,在这里面建虚拟环境。每个人的envs文件在的位置不一样,这只是我自己的位置。为什么要在这里面建虚拟环境,因为我的C盘容量不够了,在e盘建的虚拟环境,下载的pytorch也在e盘。
在这里插入图片描述
创建虚拟环境,名字是pytorch-GPU,python版本是3.8

#创建环境
conda create -n pytorch-GPU python=3.8
#进入环境
conda activate pytorch-GPU
#只输入上面两条,后面的是我自己做个记录用
#退出环境
conda deactivate
#删除环境
conda remove -n 环境名 --all

创建完后进入虚拟环境,之后依此输入以下几行

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

输入指令查看是否配置成功

conda config --show channels

显示以下
在这里插入图片描述

接下来几步很关键,不这么走的话,很可能下载的pytorch是cpu版本的,而我们要下的gpu版本的

第一步,在电脑的cmd中(不是anaconda中)输入命令nvidia-smi,查看自己电脑的cuda版本,可以看到我的版本是10.1。这里有个崩溃的小细节,昨天我看还是10.2,今天就变成10.1了,崩溃,想鼠。。。
在这里插入图片描述
第二步,根据自己电脑的cuda版本和python版本,选择pytorch版本,可以看到我是10.1版本,所以选择了1.7.0版本。
在这里插入图片描述
第三步,打开这个清华镜像链接查看自己需要的pytorch版本是否有,仔细检查你自己的python版本-pytorch版本-cuda版本在这个镜像链接里是否有。可以看到我需要的是 pytorch=1.7|python=3.8|cuda=10.1,镜像里是有这个,所以你后面下载的pytorch就是个。如果发现找不到匹配的,那么你继续操作下载的基本就是cpu版本了。具体原因以及如果没有你想要的版本组合可以看下这位大佬的解决办法(https://windses.blog.csdn.net/article/details/125910538)。当然如果你是下载完后发现下载的是cpu版的,也可以删除环境,重建按本篇再来一遍
在这里插入图片描述
第四步,去pytorch官网找下载命令。如果首页里有你的版本,就直接复制下载链接,在anaconda的命令框里输入指令下载就好,注意不要加上-c及后面的东西,加上-c是从官网下载,不加-c才是从镜像下载
在这里插入图片描述
如果没有的话选择下面那个Previous version of Pytorch,去历史版本里面找
wheel是cpu版本,不要找到下面去了,看conda部分的就可以,然后再检查一遍这个命令里的python版本-pytorch版本-cuda版本是否和镜像里的版本对应正确。
在这里插入图片描述
检查后是正确的,复制到anaconda的命令框里,记得不要加上-c及后面的东西

在这里插入图片描述
后面会出现这个,如果pytorch里显示是白线部分的那说明你下的gpu版本,输入y下载即可在这里插入图片描述
错误示范如下,如果这个链接里是有cpu,说明你下的是cpu版的,后面就别输入y了,输入n取消下载,再检查一遍版本是不是匹配错了在这里插入图片描述
下载完成后,再检查一遍下载的是否是gpu版本,在虚拟环境中输入python,导入torch包,依此输入命令torch._version_和torch.cuda.is_available()。分别得到torch版本和True说明下载成功
在这里插入图片描述
其它出现的错误
我最开始python版本用的是3.9,然后出现了一个很奇怪的报错
UnsatisfiableError: The following specifications were found
to be incompatible with the existing python installation in your environment:
在这里插入图片描述
大致意思是python版本和torch版本不匹配不兼容(?应该是)。
解决办法:重新建了一个环境并且降低了python和pytorch的版本,从python=3.9|pytorch=1.7.1降低到了python=3.8|pytorch=1.7.0,之后就成功了。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值