以下是在Debian上部署PyTorch的步骤:
更新系统并安装基础依赖
sudo apt update && sudo apt upgrade -y
sudo apt install python3 python3-pip -y # 安装Python和pip
创建虚拟环境(可选)
python3 -m venv pytorch-env
source pytorch-env/bin/activate
安装PyTorch
pip install torch torchvision torchaudio
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 # 以CUDA 11.7为例
验证安装
python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"
# 若安装CUDA版本,应输出True
安装其他依赖(可选)
pip install numpy pandas matplotlib # 根据项目需求安装
说明:
export PATH=/usr/local/cuda/bin:$PATH)。nvidia-smi查看已安装的CUDA版本。