import torch
print(torch.cuda.is_available())  # 如果返回True，说明可以使用GPU
print(torch.__version__)