pytorch 查看cuda 版本

由於pytorch的whl 安裝包名字都一樣,所以我們很難區分到底是基於cuda 的哪個版本。

有一條指令可以查看

import torch

print(torch.version.cuda)

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