anaconda安装pytorch gpu

安装PyTorch GPU版需要在Anaconda中执行以下步骤：

1. 打开Anaconda Navigator，进入"Environments"选项卡。
2. 创建一个新环境并命名，例如"pytorch-gpu"。
3. 在新环境中安装CUDA工具包，这是PyTorch GPU版所必需的。可以通过conda或者在NVIDIA官网下载安装包安装。
4. 在新环境中安装PyTorch GPU版，可以通过conda或者pip安装。例如，使用conda安装最新的PyTorch GPU版本，可以在Anaconda Prompt中输入以下命令：

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

其中，`<CUDA版本号>`应替换为已安装的CUDA版本号。如果使用pip安装，则可以执行以下命令：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu<CUDA版本号>/torch_stable.html

同样，`<CUDA版本号>`也需要替换为已安装的CUDA版本号。

5. 安装完成后，可以在新环境中打开Jupyter Notebook，导入PyTorch并使用GPU进行计算。例如，可以执行以下代码测试是否已成功安装PyTorch GPU版：

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

如果输出结果为True，则表示已成功安装PyTorch GPU版，并且可以在GPU上运行PyTorch代码了。