使用pytorch查看设备显卡详细信息

1、设置可用显卡

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3'

2、查看当前设备环境可用显卡数量

ng = torch.cuda.device_count()
print("Devices:%d" %ng)

3、查看可用显卡的具体信息（型号、算力，显存以及线程数）

import torch
infos = [torch.cuda.get_device_properties(i) for i in range(ng)]