Linux PyTorch安裝步驟是什么

在Linux系統下安裝PyTorch可以分為幾個主要步驟，具體取決于你是否需要使用GPU支持。以下是詳細的安裝步驟：

使用Anaconda安裝PyTorch（推薦）

1. 更新系統包：

   sudo apt update && sudo apt upgrade -y
   

2. 安裝Anaconda：

   • 訪問Anaconda官網，下載適用于Linux的Anaconda安裝包。
   • 在終端中導航到下載目錄，運行安裝腳本：

     bash Anaconda3-2022.10-Linux-x86_64.sh
     

   • 按照屏幕上的指示完成安裝，并在.bashrc文件中添加Anaconda的路徑。

3. 創建虛擬環境：

   conda create --name pytorch_env python=3.8
   conda activate pytorch_env
   

4. 安裝PyTorch：

   • CPU版本：

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • GPU版本（確保已安裝CUDA和cuDNN）：

     conda install pytorch torchvision torchaudio pytorch-cuda11.8 -c pytorch -c nvidia
     

5. 驗證安裝：

   • 在Python環境中輸入：

     import torch; print(torch.__version__)
     

   • 檢查CUDA和cuDNN支持（如果使用GPU）：

     import torch
     print(torch.cuda.device_count())  # 查看可用的GPU個數
     print(torch.cuda.is_available())  # 檢查CUDA是否可用
     

使用pip安裝PyTorch

1. 更新包管理器：

   sudo apt update
   

2. 安裝Python和基本庫：

   sudo apt install -y python3 python3-pip
   

3. 創建虛擬環境（可選但推薦）：

   python3 -m venv pytorch_env
   source pytorch_env/bin/activate
   

4. 安裝PyTorch：

   • CPU版本：

     pip3 install torch torchvision torchaudio
     

   • GPU版本（確保已安裝CUDA和cuDNN）：

     pip3 install torch torchvision -f https://download.pytorch.org/whl/cu118/torch_stable.html
     

5. 驗證安裝：

   • 在Python環境中輸入：

     import torch; print(torch.__version__)
     

   • 檢查CUDA和cuDNN支持（如果使用GPU）：

     import torch
     print(torch.cuda.device_count())  # 查看可用的GPU個數
     print(torch.cuda.is_available())  # 檢查CUDA是否可用
     

通過以上步驟，你應該能夠在Linux系統下成功安裝并運行PyTorch。如果遇到任何問題，可以參考PyTorch官方文檔獲取更多信息。