溫馨提示×

如何在Debian上搭建PyTorch開發環境

小樊
43
2025-08-26 10:51:49
欄目: 智能運維

在Debian上搭建PyTorch開發環境可按以下步驟進行:

  1. 更新系統并安裝基礎依賴

    sudo apt update && sudo apt upgrade -y  
    sudo apt install python3 python3-pip build-essential cmake git -y  
    
  2. 創建虛擬環境(可選)

    python3 -m venv pytorch_env  
    source pytorch_env/bin/activate  
    
  3. 安裝PyTorch

    • CPU版本
      pip install torch torchvision torchaudio  
      
    • GPU版本(需先安裝CUDA/cuDNN)
      pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118  # 替換為實際CUDA版本  
      
  4. 驗證安裝

    python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"  
    
  5. 安裝其他依賴(可選)

    pip install numpy pandas matplotlib  # 根據項目需求添加  
    

說明

  • 若使用GPU,需先安裝NVIDIA驅動、CUDA Toolkit和cuDNN,并配置環境變量(如PATHLD_LIBRARY_PATH)。
  • 推薦使用conda管理環境(尤其適合復雜項目),安裝命令可參考PyTorch官網。
  • 安裝時需根據系統CUDA版本選擇對應命令,可通過nvidia-smi查看GPU及CUDA信息。

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