以下是Debian上PyTorch的常用命令,涵蓋安裝、驗證及環境管理:
sudo apt update && sudo apt upgrade -ysudo apt install python3 python3-pippip install torch torchvision torchaudiopip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117cu117替換為實際CUDA版本,如cu102、cu120等)conda create -n pytorch_env python=3.9 -y
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch # 替換CUDA版本
python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"python3 -m venv pytorch_envsource pytorch_env/bin/activatedeactivatepip uninstall torch torchvision torchaudiopip list說明:安裝時需根據系統CUDA版本選擇對應命令,可通過nvcc --version查看CUDA版本。推薦使用虛擬環境隔離不同項目依賴。