检查自己电脑GPU个数

import torch
n_gpu = torch.cuda.device_count()

print(n_gpu)