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)了!