import torch # 查看版本 print(torch.__version__) # 查看GPU是否可用 print(torch.cuda.is_available()) # 查看GPU型号 if torch.cuda.is_available(): print(torch.cuda.get_device_name())