debian

Debian系统支持PyTorch吗

小樊
51
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
看了该问题的人还看了