import torch
torch.cuda.is_available() # cuda是否可用
True
torch.cuda.device_count() # gpu數(shù)量
9
torch.cuda.current_device() # 當(dāng)前設(shè)備索引, 從0開(kāi)始
0
torch.cuda.get_device_name(0) # 返回gpu名字
'Quadro P620'
torch.cuda.get_device_name(1)
'Tesla K80'
看看我的服務(wù)器: