import torchtorch.cuda.is_available()
import torch
torch.cuda.is_available()
查看GPU型号import torchprint(torch.cuda.get_device_name(0))
print(torch.cuda.get_device_name(0))