Linux下如何安裝PyTorch

在Linux系統上安裝PyTorch，你可以選擇使用pip或conda包管理器。以下是兩種常見的安裝方法：

使用pip安裝PyTorch

1. 打開終端： 打開你的Linux終端。

2. 更新pip： 確保你的pip是最新版本，可以使用以下命令更新pip：

   pip install --upgrade pip
   

3. 安裝PyTorch： 根據你的CUDA版本和系統架構，選擇合適的PyTorch安裝命令。你可以在PyTorch官方網站的安裝指南中找到這些命令。以下是一些常見的安裝命令示例：

   • 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.6版本：

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

   • CUDA 11.7版本：

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

   • CUDA 12.0版本：

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

使用conda安裝PyTorch

1. 打開終端： 打開你的Linux終端。

2. 更新conda： 確保你的conda是最新版本，可以使用以下命令更新conda：

   conda update conda
   

3. 創建新的conda環境（可選）： 如果你不想在當前環境中安裝PyTorch，可以創建一個新的conda環境：

   conda create -n pytorch_env python=3.9
   conda activate pytorch_env
   

4. 安裝PyTorch： 根據你的CUDA版本和系統架構，選擇合適的PyTorch安裝命令。你可以在PyTorch官方網站的安裝指南中找到這些命令。以下是一些常見的安裝命令示例：

   • CPU版本：

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • CUDA 11.3版本：

     conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
     

   • CUDA 11.6版本：

     conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch
     

   • CUDA 11.7版本：

     conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
     

   • CUDA 12.0版本：

     conda install pytorch torchvision torchaudio cudatoolkit=12.0 -c pytorch
     

驗證安裝

安裝完成后，你可以通過以下命令驗證PyTorch是否安裝成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果使用GPU版本，應該返回True

如果一切正常，你應該能夠看到PyTorch的版本號，并且如果使用的是GPU版本，torch.cuda.is_available()應該返回True。

希望這些步驟能幫助你在Linux系統上成功安裝PyTorch！