溫馨提示×

Linux系統下PyTorch的安裝教程

小樊
31
2025-09-10 19:55:13
欄目: 智能運維

以下是在Linux系統下安裝PyTorch的教程:

準備工作

更新系統包:

sudo apt update && sudo apt upgrade

安裝必要依賴:

sudo apt install -y build-essential cmake git wget unzip yasm pkg-config libopenblas-dev liblapack-dev libjpeg-dev libpng-dev

安裝Python和pip:

sudo apt install -y python3 python3-pip

創建虛擬環境(可選)

python3 -m venv pytorch-env
source pytorch-env/bin/activate

安裝PyTorch

  • CPU版本
pip install torch torchvision torchaudio
  • GPU版本(需先安裝CUDA和cuDNN)
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118  # 替換為實際CUDA版本號

也可使用conda安裝(推薦):

  • CPU版本
conda install pytorch torchvision torchaudio cpuonly -c pytorch
  • GPU版本
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia  # 替換為實際CUDA版本號

驗證安裝

python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

配置環境變量(可選)

若使用系統級安裝或需指定CUDA路徑,可在~/.bashrc~/.zshrc中添加:

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

然后運行:

source ~/.bashrc  # 或 source ~/.zshrc

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女