溫馨提示×

Debian上如何搭建PyTorch開發環境

小樊
51
2025-09-01 11:24:31
欄目: 智能運維

在Debian上搭建PyTorch開發環境可參考以下步驟:

系統準備

更新系統包:sudo apt update && sudo apt upgrade -y。

安裝Python和pip

安裝Python 3和pip3:sudo apt install python3 python3-pip -y。

創建虛擬環境(可選)

使用venv模塊創建虛擬環境:python3 -m venv pytorch_env,然后激活環境:source pytorch_env/bin/activate。

安裝PyTorch

  • CPU版本pip install torch torchvision torchaudio。
  • GPU版本:需先安裝NVIDIA驅動、CUDA和cuDNN,然后根據CUDA版本選擇安裝命令,如CUDA 11.7: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可用則為安裝成功。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女