查看pytorch的cuda版本

您可以使用以下代码查看您的PyTorch是否支持CUDA以及CUDA的版本：

import torch
if torch.cuda.is_available():
    print('CUDA version:', torch.version.cuda)
else:
    print('CUDA is not available')