如何在Ubuntu上安裝PyTorch

在Ubuntu上安裝PyTorch的步驟如下：

方法一：使用pip安裝

1. 更新系統包列表

   sudo apt update
   

2. 安裝Python和pip 如果你還沒有安裝Python和pip，可以使用以下命令：

   sudo apt install python3 python3-pip
   

3. 安裝PyTorch 根據你的CUDA版本選擇合適的PyTorch安裝命令。你可以在PyTorch官網找到最新的安裝命令。

   • 無GPU支持（CPU版本）

     pip3 install torch torchvision torchaudio
     

   • 有GPU支持（CUDA 11.7）

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

   • 其他CUDA版本 請參考PyTorch官網上的相應CUDA版本安裝命令。

4. 驗證安裝 打開Python解釋器并嘗試導入PyTorch：

   python3
   >>> import torch
   >>> print(torch.__version__)
   

方法二：使用conda安裝

如果你更喜歡使用conda來管理包和環境，可以按照以下步驟操作：

1. 安裝Miniconda或Anaconda 你可以從Miniconda官網或Anaconda官網下載并安裝Miniconda或Anaconda。

2. 創建新的conda環境

   conda create -n pytorch_env python=3.9
   

3. 激活環境

   conda activate pytorch_env
   

4. 安裝PyTorch 根據你的CUDA版本選擇合適的PyTorch安裝命令。你可以在PyTorch官網找到最新的安裝命令。

   • 無GPU支持（CPU版本）

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • 有GPU支持（CUDA 11.7）

     conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
     

   • 其他CUDA版本 請參考PyTorch官網上的相應CUDA版本安裝命令。

5. 驗證安裝 打開Python解釋器并嘗試導入PyTorch：

   python
   >>> import torch
   >>> print(torch.__version__)
   

注意事項

• 確保你的系統已經安裝了CUDA Toolkit（如果你打算使用GPU版本）。你可以從NVIDIA官網下載并安裝CUDA Toolkit。
• 安裝過程中可能會遇到依賴沖突或其他問題，請根據錯誤提示進行相應的調整。

通過以上步驟，你應該能夠在Ubuntu上成功安裝PyTorch。