問題
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之間的匹配。