Debian系统支持PyTorch。可通过pip或conda安装,安装时需根据系统是否支持GPU选择对应版本。
安装步骤:
sudo apt update && sudo apt upgradesudo apt install python3 python3-pippip install torch torchvision torchaudiopip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117(以CUDA 11.7为例)conda create -n pytorch_env python=3.9conda activate pytorch_envconda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorchimport torch print(torch.__version__) print(torch.cuda.is_available()) # 若安装GPU版本,应返回True 注:安装前需确保系统已安装CUDA(若使用GPU)及对应驱动,可参考PyTorch官网指南。