1.py

import torch
print(f"CUDA Available: {torch.cuda.is_available()}")
print(f"Device Name: {torch.cuda.get_device_name(0)}")
import torch
print("GPU是否可用:", torch.cuda.is_available())
print("显卡型号:", torch.cuda.get_device_name(0))