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))
参考:
添加链接描述
添加链接描述