import torch
print("Torch 是否使用 GPU:", torch.cuda.is_available())
print("当前使用的 GPU:", torch.cuda.get_device_name(0) if torch.cuda.is_available() else "CPU")