torch.cuda.is_available() 返回 False

安装pytorch总会遇到这样那样的问题,导致要么import torch出错,要么torch.cuda.is_available()输出False。今天把我遇到的所有问题整理了一个简单的总结,仅供参考。

1 显卡

首先确保你的显卡支持CUDA。 (不展开,自己面对百度)

2 驱动

驱动最好是升级到最新。我是下载了这个:GeForce Experience,然后升级里面的GeForce Game Ready到最新就可以。
torch.cuda.is_available() 返回 False

3 CPU or GPU

如果以上两个都没有问题,可以通过Anaconda Promptconda activate 环境名进入你的环境,然后检查你安装的pytorch是否有cpuonlyconda list,因为长安装时有gpugpu版本。
如果有cpuonly字,可以用conda uninstall cpuonly(或pip)卸载。
如果不适合你,建议去官网重新下载pytorch。通过pip wheel命令下载一般是没有问题的。这就是我解决它的方法。

4 版本

如果你之前安装并成功调用了cuda,但是你的pytorch版本由于某种原因发生了变化(有些包是捆绑下载的,下载后会自动更改pytorch版本),你就不能再调用GPU了。这个时候最好的解决办法是把pytorch安装回原来的版本。
如果你太过于着急,已经把cudapytorch都重装了一遍,能运行就不说了,不能运行的话,可以参考我这篇文章:OSError: [WinError 127] 找不到指定的程序、“caffe2_detectron_ops.dll“ or one of its dependencies.

5 待更新

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2022年4月10日 上午11:23
下一篇 2022年4月13日 下午6:38

相关推荐