怎么检查安装的PyTorch版本和CUDA版本是否兼容

要检查PyTorch版本和CUDA版本是否兼容，可以使用以下命令：

import torch

print(torch.version.cuda)  # 打印PyTorch使用的CUDA版本
print(torch.cuda.is_available())  # 检查CUDA是否可用

如果输出的CUDA版本与你安装的CUDA版本不同，则说明PyTorch版本和CUDA版本不兼容。在这种情况下，你需要安装一个与你的PyTorch版本兼容的CUDA版本。你可以在PyTorch官方网站上查看PyTorch和CUDA版本的兼容性矩阵。