Ver la GPU del servidor local o remoto y el método de uso basado en Pytorch

Ver la GPU del servidor local o remoto y el método de uso basado en Pytorch

1. Ver información específica de la GPU

nvidia-smi # 查看GPU型号等具体信息

2. Ver GPU de servidor local o remoto basado en Pytorch

import torch
print(torch.cuda.is_available() )# cuda是否可用
print(torch.cuda.device_count() )# gpu数量
print(torch.cuda.current_device())# 当前设备索引, 从0开始
print(torch.cuda.get_device_name(0))# 返回gpu名字

3. Cómo usar GPU (típico)

device = torch.cuda.current_device() if args.cuda else torch.device('cpu')
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
os.environ['CUDA_VISIBLE_DEVICES']='0'