torch查看GPU信息常用代码

torch.cuda.device_count() # 可以返回可用 GPU 的数量。

torch.cuda.get_device_name(device) # 可以获取特定 GPU 的名称。其中 device 是 GPU 的索引（从 0 开始）

# 查看所有gpu
for i in range(torch.cuda.device_count()):
    print(torch.cuda.get_device_name(i))
'''
Tesla T4
Tesla T4
'''