import torch

查看cuda是否可用

print(torch.cuda.is_available())

查看cuda設備的數量

print(torch.cuda.device_count())

查看當前使用的cuda編號

print(torch.cuda.current_device())

查看GPU設備名字

print(torch.cuda.get_device_name())

查看設備容量

print(torch.cuda.get_device_capability(0)) return: True 1 0 GeForce GTX 1060 (6, 1)

總結

以上是生活随笔為你收集整理的【pytorch】torch 查看GPU的全部內容，希望文章能夠幫你解決所遇到的問題。

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