使用conda安装gpu版本的pytorch

使用Conda安装GPU版本的PyTorch通常需要额外的步骤，因为PyTorch的CUDA版本依赖于特定的操作系统、CUDA库和Python版本。以下是安装过程的一个简要概述：

1. **安装 Anaconda**：
如果你还没有安装Anaconda，首先访问https://www.anaconda.com/products/distribution 下载并安装适合你的系统（包括Python 3.x版本）。

2. **创建新环境**：
打开命令行终端，输入以下命令来创建一个新的环境，指定包含GPU支持：

   conda create -n pytorch_gpu python=3.8 torchvision cudatoolkit=10.2 or your CUDA version

确保替换`10.2`为你的CUDA版本号。

3. **激活环境**：
创建完环境后，激活它：

   conda activate pytorch_gpu

4. **安装PyTorch**：
使用下面的命令来安装PyTorch及其相关库，可能会有CUDA版本的选项：

   conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

5. **检查安装**：
安装完成后，可以运行以下命令来验证PyTorch是否成功安装，并且可以访问GPU：

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

如果输出结果为True，则说明GPU已成功启用。