問題

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

解決

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

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。