以下是在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
pip install torch torchvision torchaudio
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 # 替換為實際CUDA版本號
也可使用conda安裝(推薦):
conda install pytorch torchvision torchaudio cpuonly -c pytorch
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