pytorch-CPU版本无须卸载转pytorch-GPU版本

pytorch-CPU版本无须卸载转pytorch-GPU版本
一直以来都以为自己的win装的GPU 版本的pytorch,因为一直都在用Ubuntu那台台式机,这个win只是用来做一下简单的机器学习sklearn跟深度学习作对比,后来密接隔离的时候带了笔记本,
切换pytorch环境,发现运行不了to.(device),查找很多资料,版本一直对的上,但是就是运行不了GPU。
因为里面装了很多包,不想卸载,太麻烦,就想有没有换掉其中的一些包,改成GPU版本的pytorch

首先在再conda prompt pytorch的环境中

conda uninstall pytorch torchvision torchaudio

卸载torchvision和torchaudio才发现一直装的都是CPU版本


conda uninstall cpuonly

这一步很重要,卸载cpuonely

进入pytorch官网
在这里插入图片描述
想安装最新版的pytorch-GPU,但是输入之后

请添加图片描述
还是显示CPU only
没办法只有降低版本选取

在这里插入图片描述

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

输入上面代码

请添加图片描述
可以看到有了cuda,输入y,回车安装即可

图上没有安装成功,因为外国网站比较慢,可以换源,
但是也有玄学,就是在早上安装比较快,越早越快
第二天重新安装,很快就好了
然后

在这里插入图片描述
就可以尽情to(device)了!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值