在Debian上搭建PyTorch開發環境可按以下步驟進行:
更新系統并安裝基礎依賴
sudo apt update && sudo apt upgrade -y
sudo apt install python3 python3-pip build-essential cmake git -y
創建虛擬環境(可選)
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/cu118 # 替換為實際CUDA版本
驗證安裝
python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"
安裝其他依賴(可選)
pip install numpy pandas matplotlib # 根據項目需求添加
說明:
PATH
和LD_LIBRARY_PATH
)。conda
管理環境(尤其適合復雜項目),安裝命令可參考PyTorch官網。nvidia-smi
查看GPU及CUDA信息。