pytorchgpu版本安装

您可以通过以下步骤安装 PyTorch GPU 版本：

1. 确认您的显卡支持 CUDA，并且已经安装了相应的驱动程序。
2. 安装 CUDA Toolkit，可以从 NVIDIA 官网下载对应版本的 CUDA Toolkit。
3. 安装 cuDNN，可以从 NVIDIA 官网下载对应版本的 cuDNN。
4. 打开 Anaconda Prompt 或者终端，创建一个新的虚拟环境：`conda create --name pytorch_gpu python=3.8`
5. 激活虚拟环境：`conda activate pytorch_gpu`
6. 安装 PyTorch GPU 版本：`conda install pytorch torchvision torchaudio cudatoolkit=<CUDA 版本号> -c pytorch`

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