
安装pytorch总会遇到这样那样的问题,导致要么import torch出错,要么torch.cuda.is_available()输出False。
今天我把我遇到的所有问题做一个简单的汇总,仅供参考。
首先要确认你的显卡是否支持CUDA。
(不展开,自行面向百度)
2 驱动驱动最好是升级到最新。
我是下载了这个:GeForce Experience,然后升级里面的GeForce Game Ready到最新就可以。
如果上面两个都没有问题,你可以通过Anaconda Prompt、conda activate 环境名进入你的环境,然后通过conda list检查你安装的pytorch是不是有cpuonly,因为安装pytorch时有cpu版本也有gpu版本。
如果有cpuonly的字样,可以用conda uninstall cpuonly(或pip)卸载,这是我翻别人博客发现的方法,对我没有用。
如果对你也没用,建议重新去官网下载pytorch,通过pip wheel那个命令下载一般没问题,我是这样解决的。
如果你之前安装并成功调用cuda了,但是由于某种原因你的pytorch版本被改变了(有些包是捆绑下载,下载了之后会自动改变pytorch版本),不能再调用GPU了。
这时候最好的解决方法是把pytorch装回原先的版本。
如果你太过于着急,已经把cuda和pytorch都重装了一遍,能运行就不说了,不能运行的话,可以参考我这篇文章:OSError: [WinError 127] 找不到指定的程序、“caffe2_detectron_ops.dll“ or one of its dependencies.
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)