anaconda环境下安装GPU版本的pytorch

您可以按照以下步骤在Anaconda环境下安装GPU版本的PyTorch：

1. 首先，确保您已经安装了NVIDIA显卡以及相应的驱动程序和CUDA工具包。

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

conda create --name gpu_env
conda activate gpu_env

3. 安装PyTorch和torchvision：

conda install pytorch torchvision torchaudio cudatoolkit=<version>

其中，`<version>`应该替换为您安装的CUDA工具包版本号。例如，如果您安装了CUDA 11.1，则应该使用以下命令：

conda install pytorch torchvision torchaudio cudatoolkit=11.1

4. 等待安装完成后，您可以在Python中导入PyTorch并验证GPU是否可用：

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

如果输出为True，则表示GPU已成功启用。

希望这些步骤能够帮助您成功安装GPU版本的PyTorch。