在windows安装gpu版的pytorch

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

1. 安装CUDA工具包：访问NVIDIA官网下载并安装CUDA工具包，选择适合自己系统的版本。

2. 安装cuDNN库：访问NVIDIA官网下载并安装cuDNN库，选择与自己系统和CUDA版本相匹配的版本。

3. 安装Anaconda：访问Anaconda官网下载并安装Anaconda，选择适合自己系统的版本。

4. 创建虚拟环境：在Anaconda Prompt中运行以下命令创建虚拟环境：

   conda create -n pytorch_gpu python=3.7

5. 激活虚拟环境：在Anaconda Prompt中运行以下命令激活虚拟环境：

   conda activate pytorch_gpu

6. 安装PyTorch：在Anaconda Prompt中运行以下命令安装PyTorch：

   conda install pytorch torchvision torchaudio cudatoolkit=<CUDA版本号> -c pytorch

其中，<CUDA版本号>需要替换为自己安装的CUDA版本号，如cudatoolkit=11.1。

7. 验证安装：在Anaconda Prompt中运行以下命令验证是否安装成功：

   python -c "import torch; print(torch.cuda.is_available())"

如果输出True，则表示安装成功。

注意事项：

- 安装过程中需要联网下载并安装相关依赖包，需要保持网络畅通。
- 安装时需要注意CUDA和cuDNN的版本要与PyTorch版本匹配，否则会导致安装失败或无法正常运行。