在Debian系統解決PyTorch依賴問題可按以下步驟操作:
更新系統包
sudo apt update && sudo apt upgrade -y
安裝基礎依賴
sudo apt install -y python3 python3-pip build-essential cmake git libopenblas-dev liblapack-dev libjpeg-dev libpng-dev
創建虛擬環境(可選)
python3 -m venv pytorch-env
source pytorch-env/bin/activate
安裝PyTorch
pip install torch torchvision torchaudio
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 # 替換為對應CUDA版本
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch # 自動處理依賴
驗證安裝
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 檢查GPU支持
常見問題:
pip install --upgrade
升級沖突包,或通過虛擬環境隔離。參考來源: