Windows pytorch 切换cuda版本

要在 Windows 上切换 PyTorch 的 CUDA 版本，可以按照以下步骤进行操作：

1. 首先卸载当前 CUDA 版本和与之相关的 PyTorch 版本。

2. 安装新的 CUDA 版本，可以在 NVIDIA 官网上下载相应的版本。

3. 安装与新 CUDA 版本相对应的 cuDNN 库，同样可以在 NVIDIA 官网上下载。

4. 重新安装 PyTorch，可以使用 pip 命令安装。在安装时，需要指定新 CUDA 版本的路径，例如：

pip install torch==1.7.1+cu110 -f https://download.pytorch.org/whl/cu110/torch_stable.html

其中，cu110 表示 CUDA 11.0 版本，如果是其他版本，需要相应地更改。

5. 验证安装是否成功，可以在 Python 中输入以下代码：

import torch
print(torch.version.cuda)

如果输出的版本号是新安装的 CUDA 版本，则表示切换成功。

注意：在切换 CUDA 版本时，需要保证新旧版本的兼容性，否则可能会出现不兼容的情况。建议在切换前先查看 PyTorch 官方文档中的版本兼容性说明。