Debian上安裝PyTorch失敗可按以下步驟排查解決:
更新系統并安裝基礎依賴
sudo apt update && sudo apt upgrade -y
sudo apt install -y python3 python3-pip build-essential cmake git libopenblas-dev liblapack-dev
檢查Python和pip版本
確保Python≥3.6,pip≥21.0,可通過python3 --version和pip3 --version查看,低版本需升級:
sudo apt install -y 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中運行以下代碼,檢查是否成功及CUDA是否可用:
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 若為True則GPU支持正常
其他常見問題
nvidia-smi查看GPU支持的CUDA版本,再選擇對應PyTorch版本。python3 -m venv pytorch_env
source pytorch_env/bin/activate
pip install torch torchvision torchaudio
```。
若仍失敗,建議查看錯誤日志,或參考PyTorch官方文檔及社區論壇(如Stack Overflow、Reddit)尋求具體解決方案。