問題

1.執(zhí)行Nvidia-smi 命令沒有報錯，能夠顯示驅(qū)動信息；
2.執(zhí)行 torch.backends.cudnn.enabled is TRUE
3.torch.cuda.is_available()一直返回False

解決

把torch，torchvision等相關(guān)安裝包全部刪除，安裝適合版本的torch。

30系列顯卡是新一代架構(gòu)，新驅(qū)動不支持cuda9以及cuda10，所以必須安裝cuda11、而pytorch現(xiàn)在穩(wěn)定版為1.6，最高僅支持到cud10.2。所以唯一的辦法就是使用上處于beta測試的1.7或1.8。這也是為啥一開始就強調(diào)本文的寫作時間，因為未來一定會有穩(wěn)定版發(fā)布，請使用穩(wěn)定版。torch_geometric 計算包官方發(fā)布的現(xiàn)成輪子僅支持到pytorch-1.6和cuda10.2。
所以核心問題就是cuda11 pytorch 以及torch_geometric之間的匹配。

總結(jié)

以上是生活随笔為你收集整理的3090显卡 torch.cuda.is_available()返回false的解决办法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。