20201114-利用torch查看GPU

import torch
torch.cuda.is_available() # cuda是否可用

True

torch.cuda.device_count() # gpu数量

9

torch.cuda.current_device() # 当前设备索引, 从0开始

0

torch.cuda.get_device_name(0) # 返回gpu名字

'Quadro P620'

torch.cuda.get_device_name(1)

'Tesla K80'

看看我的服务器：