问题:torch.cuda.is_available()返回false

搞死我了一整天!  torch.cuda.is_available()总是返回False。

周一来上班一开电脑居然cuda用不了,上周五电脑没关机只是休眠,代码还跑的好好。

一整天到下班还没整好,尝试了各种办法,pytorch, cuda,cudnn 各种版本换来换去都不行。

后来看到可能是cuda10的版本比较高驱动器不匹配,就决定升级驱动。 

最后换了驱动器,把390换成了430,直接在ubuntu的软件更新那里更换驱动器,然后重启,

torch.cuda.is_available() 打印出来是True

 

我的版本对应如下

 

print(torch.__version__ )  === 1.2.0
print(torch.version.cuda) ==== 10.0.130
print(torch.cuda.is_available()) == True

 

你可能感兴趣的:(小问题大烦恼)