溫馨提示×

Linux系統PyTorch安裝步驟是什么

小樊
47
2025-09-14 19:52:46
欄目: 智能運維

以下是在Linux系統上安裝PyTorch的步驟:

安裝Python和pip

大多數Linux發行版默認安裝了Python,可使用以下命令檢查:

python3 --version
pip3 --version

若未安裝,對于Debian系(如Ubuntu),可執行:

sudo apt update
sudo apt install python3 python3-pip

對于RPM系(如CentOS),可執行:

sudo yum install python3 python3-pip

創建虛擬環境(可選)

為避免依賴沖突,建議創建虛擬環境:

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

安裝PyTorch

  • 使用pip安裝
    • CPU版本pip install torch torchvision torchaudio
    • GPU版本(需先安裝CUDA和cuDNN)pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118(將cu118替換為實際CUDA版本號)
  • 使用conda安裝
    • 安裝Miniconda(若未安裝)
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
- **創建并激活環境**:
conda create -n pytorch-env python=3.8
conda activate pytorch-env
- **安裝PyTorch**:
    - **CPU版本**:`conda install pytorch torchvision torchaudio cpuonly -c pytorch`
    - **GPU版本**:`conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia`(將`11.8`替換為實際CUDA版本號)

驗證安裝

在Python環境中運行以下代碼:

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 若為GPU版本,應返回True

配置環境變量(可選)

若使用GPU,需配置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
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女