以下是在Linux系統上安裝PyTorch的步驟:
大多數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
pip install torch torchvision torchaudio
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
(將cu118
替換為實際CUDA版本號)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