查看pytorch對應的cuda版本

進入python編譯環境,

輸入import torch 回車,

再輸入:print(torch.version.cuda)。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章