温馨提示×

PyTorch在Debian上的部署方法

小樊
61
2025-09-01 11:26:36
栏目: 智能运维

以下是在Debian上部署PyTorch的步骤:

  1. 更新系统并安装基础依赖

    sudo apt update && sudo apt upgrade -y  
    sudo apt install python3 python3-pip -y  # 安装Python和pip  
    
  2. 创建虚拟环境(可选)

    python3 -m venv pytorch-env  
    source pytorch-env/bin/activate  
    
  3. 安装PyTorch

    • CPU版本
      pip install torch torchvision torchaudio  
      
    • CUDA版本(需先安装NVIDIA驱动、CUDA Toolkit和cuDNN):
      pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117  # 以CUDA 11.7为例  
      
  4. 验证安装

    python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"  
    # 若安装CUDA版本,应输出True  
    
  5. 安装其他依赖(可选)

    pip install numpy pandas matplotlib  # 根据项目需求安装  
    

说明

  • 若使用GPU,需先安装NVIDIA驱动、CUDA和cuDNN,并设置环境变量(如export PATH=/usr/local/cuda/bin:$PATH)。
  • 安装命令中的CUDA版本需与本地环境匹配,可通过nvidia-smi查看已安装的CUDA版本。

0