Debian系统安装PyTorch失败可按以下步骤排查解决:
检查基础环境
sudo apt update && sudo apt upgrade -y
sudo apt install -y build-essential cmake git python3-dev python3-pip
python3 --version
、pip3 --version
,确保为Python 3.x安装PyTorch
conda create -n pytorch_env python=3.8
conda activate pytorch_env
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia
pip3 install torch torchvision torchaudio
pip3 install torch torchvision -f https://download.pytorch.org/whl/cu118/torch_stable.html
(需匹配CUDA版本)解决常见问题
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch
nvcc --version
查看CUDA版本。--user
参数安装到用户目录,或使用sudo
。验证安装
运行Python代码检查:
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 若为GPU版本,应返回True
若仍失败,建议查看错误日志,搜索具体错误信息(如“GLIBC版本不兼容”等),或在PyTorch官方论坛/GitHub提交问题。