【pytorch】torch 查看GPU
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)
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！