anconda安装pytorch gpu

可以按照以下步骤安装 pytorch gpu 版本：

1. 检查你的电脑是否支持 CUDA，可以通过运行以下命令检查：

   nvcc --version

如果输出类似以下内容，说明你的电脑支持 CUDA：

   nvcc: NVIDIA (R) Cuda compiler driver
   Copyright (c) 2005-2019 NVIDIA Corporation
   Built on Sun_Jul_28_19:07:16_PDT_2019
   Cuda compilation tools, release 10.1, V10.1.243

2. 安装 Anaconda 或 Miniconda（推荐使用 Miniconda）。

3. 创建一个新的虚拟环境（可选），例如：

   conda create --name pytorch
   conda activate pytorch

4. 安装 pytorch gpu 版本。可以在官网上找到对应的安装命令，例如：

   conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

5. 验证 pytorch 是否成功安装。可以通过运行以下代码验证：

   import torch

   if torch.cuda.is_available():
       device = torch.device("cuda")
   else:
       device = torch.device("cpu")

   print(device)

如果输出为 "cuda"，说明 pytorch gpu 版本安装成功。