在Debian系統選擇PyTorch版本時,需結合硬件配置和項目需求,具體步驟如下:
確認硬件環境
pip install torch torchvision torchaudio
。pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
。檢查Python版本兼容性
python3 --version
查看。pip install --upgrade pip
。推薦安裝方式
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
。驗證安裝
安裝后運行以下代碼,檢查版本及GPU是否可用:
import torch
print(torch.__version__) # 查看版本
print(torch.cuda.is_available()) # 若為True則GPU可用
注意事項:
nvidia-smi
查看GPU型號及驅動版本。