windows怎么安装gpu版本的pytorch

要在 Windows 上安装 GPU 版本的 PyTorch，需要执行以下步骤：

1. 首先，在你的机器上安装 CUDA 工具包。你可以从 NVIDIA 的官方网站上下载和安装适合你的操作系统版本的 CUDA。确保你下载的是与你的 GPU 兼容的版本。

2. 安装 cuDNN 库。cuDNN 是 NVIDIA 提供的一个深度神经网络加速库，其支持 CUDA 加速。你可以从 NVIDIA 开发者网站下载适合你的操作系统版本的 cuDNN 库。

3. 安装 Anaconda 或 Miniconda。如果你已经安装了这些软件，请跳过此步骤。

4. 打开 Anaconda Prompt 或者命令行工具，输入以下命令：

conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version>

其中，`<your_cuda_version>` 应该被替换为你所安装的 CUDA 版本号。例如，如果你安装了 CUDA 11.1，则应该输入：

conda install pytorch torchvision torchaudio cudatoolkit=11.1

如果你需要安装特定版本的 PyTorch，可以在命令中指定版本号，例如：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1

5. 等待安装完成后，即可在 Python 中导入 PyTorch 库并开始使用 GPU 加速了。