import torch
print(torch.cuda.is_available())
一般都是在命令行裏：
返回True說明GPU可用
pytorch之torch.cuda.is_available()——判斷GPU可用
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
import torch
print(torch.cuda.is_available())
一般都是在命令行裏：
返回True說明GPU可用