查看pytorch gpu版本

import torch
torch.cuda.is_available()

查看GPU型号

import torch
print(torch.cuda.get_device_name(0))