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)寻求具体解决方案。