linux anaconda 安装pytorch-gpu

在Linux上安装PyTorch-GPU需要经过以下步骤：

1. 首先安装Anaconda，可以从官网下载安装包并按照提示进行安装。

2. 打开终端或控制台，使用conda创建一个新的虚拟环境：

   conda create -n env_pytorch python=3.7

这里我们取名为env_pytorch。

3. 激活虚拟环境并安装PyTorch-GPU：

   conda activate env_pytorch
   conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

这里使用的是PyTorch的官方源和conda-forge源，同时指定了CUDA的版本号。

4. 安装完成后，可以在Python中测试是否成功安装了PyTorch-GPU：

   python
   import torch

   print(torch.cuda.is_available()) # 输出True表示可以使用GPU

如果输出True，则说明已经成功安装了PyTorch-GPU。

注意事项：

- PyTorch的GPU版本需要与CUDA的版本对应，因此需要确定所安装的CUDA版本。
- 如果使用conda-forge源，则需要先将其添加到Anaconda中：

  conda config --add channels conda-forge