torch.cuda.is

torch.cuda.is,第1张

安装pytorch总会遇到这样那样的问题,导致要么import torch出错,要么torch.cuda.is_available()输出False


今天我把我遇到的所有问题做一个简单的汇总,仅供参考。


1 显卡

首先要确认你的显卡是否支持CUDA


(不展开,自行面向百度)

2 驱动

驱动最好是升级到最新。


我是下载了这个:GeForce Experience,然后升级里面的GeForce Game Ready到最新就可以。



3 CPU or GPU

如果上面两个都没有问题,你可以通过Anaconda Promptconda activate 环境名进入你的环境,然后通过conda list检查你安装的pytorch是不是有cpuonly,因为安装pytorch时有cpu版本也有gpu版本。



如果有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 待更新

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/578390.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-11
下一篇 2022-04-11

发表评论

登录后才能评论

评论列表(0条)

保存