# 测试是否有gpu
import torch
print(torch.cuda.is_available())