Anaconda安装GPU版本的pytorch

您可以通过以下步骤在Anaconda中安装GPU版本的PyTorch：

1. 首先，确保您的计算机上已经安装了NVIDIA GPU驱动程序和CUDA工具包。您可以在NVIDIA官网上下载并安装它们。

2. 打开Anaconda Prompt，创建一个新的虚拟环境并激活它：

   conda create --name pytorch_gpu
   conda activate pytorch_gpu

3. 安装PyTorch和torchvision：

   conda install pytorch torchvision torchaudio cudatoolkit=<version>

其中，`<version>`是您安装的CUDA工具包的版本号。例如，如果您安装的是CUDA 11.1，则应该使用`cudatoolkit=11.1`。

4. 安装其他必要的依赖项：

   conda install matplotlib numpy jupyter

5. 测试安装是否成功：

   python
   import torch
   print(torch.cuda.is_available())

如果输出为True，则说明您已经成功安装了GPU版本的PyTorch。