溫馨提示×

Debian系統支持PyTorch嗎

小樊
49
2025-08-10 05:34:58
欄目: 智能運維

Debian系統支持PyTorch??赏ㄟ^pip或conda安裝,安裝時需根據系統是否支持GPU選擇對應版本。
安裝步驟:

  1. 更新系統sudo apt update && sudo apt upgrade
  2. 安裝Python和pipsudo apt install python3 python3-pip
  3. 選擇安裝方式
    • pip安裝
      • CPU版本:pip install torch torchvision torchaudio
      • GPU版本(需先安裝CUDA):pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117(以CUDA 11.7為例)
    • conda安裝
      • 創建虛擬環境:conda create -n pytorch_env python=3.9
      • 激活環境:conda activate pytorch_env
      • 安裝:conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
  4. 驗證安裝
    import torch  
    print(torch.__version__)  
    print(torch.cuda.is_available())  # 若安裝GPU版本,應返回True  
    

注:安裝前需確保系統已安裝CUDA(若使用GPU)及對應驅動,可參考PyTorch官網指南。

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