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-CPU版本无须卸载转pytorch-GPU版本_第1张图片
想安装最新版的pytorch-GPU,但是输入之后

pytorch-CPU版本无须卸载转pytorch-GPU版本_第2张图片
还是显示CPU only
没办法只有降低版本选取

pytorch-CPU版本无须卸载转pytorch-GPU版本_第3张图片

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

输入上面代码

pytorch-CPU版本无须卸载转pytorch-GPU版本_第4张图片
可以看到有了cuda,输入y,回车安装即可

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

pytorch-CPU版本无须卸载转pytorch-GPU版本_第5张图片
就可以尽情to(device)了!

你可能感兴趣的:(1024程序员节,python,pandas,pytorch)