安装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条)