解决PyTorch与CUDA版本不匹配问题

PyTorch是一个广泛使用的深度学习框架，而CUDA是NVIDIA提供的用于并行计算的平台和编程模型。在使用PyTorch进行深度学习任务时，经常会遇到PyTorch与CUDA版本不匹配的问题。本文将详细介绍如何解决这个问题，并提供相应的源代码示例。

1. 检查CUDA版本
   首先，我们需要确定当前系统中的CUDA版本。可以通过以下代码来检查CUDA版本：

import torch

print(torch.version.cuda)

运行以上代码，将输出当前系统中安装的CUDA版本。记下此版本号，以备后续使用。

2. 检查PyTorch版本
   接下来，我们需要检查当前安装的PyTorch版本。可以使用以下代码来检查PyTorch版本：