在Debian上搭建PyTorch開發環境可參考以下步驟:
更新系統包:sudo apt update && sudo apt upgrade -y
。
安裝Python 3和pip3:sudo apt install python3 python3-pip -y
。
使用venv
模塊創建虛擬環境:python3 -m venv pytorch_env
,然后激活環境:source pytorch_env/bin/activate
。
pip install torch torchvision torchaudio
。pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
。在Python環境中運行:import torch; print(torch.__version__); print(torch.cuda.is_available())
,若顯示版本號且CUDA可用則為安裝成功。