linux環境下pytorch安裝指南

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

使用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-cuda=11.8 -c pytorch -c nvidia
     

5. 驗證安裝

   python -c "import torch; print(torch.__version__)"
   

使用pip安裝PyTorch

1. 更新包管理器

   sudo apt-get update
   

2. 安裝Python和基本庫

   sudo apt-get install python3-pip
   pip3 install torch torchvision
   

3. 檢查CUDA和cuDNN支持（如果使用GPU）

   • 確保你的系統已經安裝了CUDA和cuDNN。
   • 安裝PyTorch GPU版本：

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

4. 驗證安裝

   python -c "import torch; print(torch.__version__)"
   

請根據你的系統配置和需求選擇合適的安裝方法。如果你使用的是GPU版本，請確保你的系統已經正確安裝了NVIDIA驅動和相應的CUDA版本。