PyTorch在Linux环境如何安装

在Linux环境下安装PyTorch，可以按照以下步骤进行：

方法一：使用pip安装

1. 更新pip：

   pip install --upgrade pip
   

2. 安装PyTorch： 根据你的CUDA版本选择合适的安装命令。以下是一些常见的CUDA版本对应的安装命令：

   • 无GPU（CPU版本）：

     pip install torch torchvision torchaudio
     

   • CUDA 11.3：

     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
     

   • CUDA 11.2：

     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu112
     

   • CUDA 11.1：

     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu111
     

   • CUDA 10.2：

     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu102
     

   • CUDA 10.1：

     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu101
     

   • CUDA 9.2：

     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu92
     

   • CUDA 8.0：

     pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu80
     

   你可以访问PyTorch官网获取最新的安装命令和详细信息。

方法二：使用conda安装

如果你使用的是Anaconda或Miniconda，可以使用conda来安装PyTorch：

1. 更新conda：

   conda update conda
   

2. 安装PyTorch： 同样根据你的CUDA版本选择合适的安装命令。以下是一些常见的CUDA版本对应的安装命令：

   • 无GPU（CPU版本）：

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • CUDA 11.3：

     conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
     

   • CUDA 11.2：

     conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch
     

   • CUDA 11.1：

     conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
     

   • CUDA 10.2：

     conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
     

   • CUDA 10.1：

     conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
     

   • CUDA 9.2：

     conda install pytorch torchvision torchaudio cudatoolkit=9.2 -c pytorch
     

   • CUDA 8.0：

     conda install pytorch torchvision torchaudio cudatoolkit=8.0 -c pytorch
     

   你可以访问PyTorch官网获取最新的安装命令和详细信息。

验证安装

安装完成后，可以通过以下代码验证PyTorch是否安装成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 检查是否有可用的GPU

如果输出显示了PyTorch的版本号，并且torch.cuda.is_available()返回True（如果你安装的是GPU版本），则说明安装成功。

希望这些步骤能帮助你在Linux环境下成功安装PyTorch！