Ubuntu中PyTorch依賴怎么安裝

在Ubuntu系統中安裝PyTorch及其依賴，可以按照以下步驟進行：

方法一：使用pip安裝

1. 更新系統包列表

   sudo apt update
   

2. 安裝必要的依賴項

   sudo apt install python3-pip python3-dev
   

3. 安裝PyTorch PyTorch官方提供了多種安裝方式，可以根據你的CUDA版本選擇合適的命令。以下是一些常見的安裝命令：

   • CPU版本

     pip3 install torch torchvision torchaudio
     

   • CUDA 11.3版本

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

   • CUDA 11.7版本

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

   • CUDA 12.0版本

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

   請根據你的GPU和CUDA版本選擇合適的命令。

方法二：使用conda安裝

如果你更喜歡使用Anaconda或Miniconda來管理Python環境和包，可以按照以下步驟進行：

1. 安裝Anaconda或Miniconda

   • 下載并安裝Anaconda或Miniconda：Anaconda下載頁面 或 Miniconda下載頁面

2. 創建新的conda環境

   conda create -n pytorch_env python=3.8
   

3. 激活環境

   conda activate pytorch_env
   

4. 安裝PyTorch 同樣，根據你的CUDA版本選擇合適的命令：

   • CPU版本

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • CUDA 11.3版本

     conda install pytorch torchvision torchaudio cudatoolkit=11.3 -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是否可用的信息。

希望這些步驟能幫助你在Ubuntu系統中成功安裝PyTorch及其依賴！