解決Debian上PyTorch安裝失敗可按以下步驟排查:
更新系統依賴
sudo apt update && sudo apt upgrade -y
sudo apt install -y build-essential cmake git libopenblas-dev liblapack-dev python3-dev python3-pip
檢查Python與pip版本
確保Python≥3.6,pip≥21.0,可通過以下命令升級:
sudo apt install -y python3 python3-pip
pip3 install --upgrade pip
選擇正確安裝命令
pip3 install torch torchvision torchaudio
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
使用國內鏡像加速
若安裝慢,可臨時指定清華鏡像源:
pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
驗證安裝
在Python中運行:
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 檢查GPU是否可用
常見問題處理
sudo
或使用虛擬環境。nvidia-smi
檢查GPU驅動是否正確安裝。若仍失敗,可參考PyTorch官方安裝指南或社區論壇排查具體錯誤信息。