温馨提示×

PyTorch在Linux环境中的安装步骤是什么

小樊
48
2025-09-06 19:33:18
栏目: 智能运维

以下是在Linux环境中安装PyTorch的步骤:

  1. 更新系统包

    sudo apt update && sudo apt upgrade  
    
  2. 安装Python和pip

    sudo apt install python3 python3-pip  
    
  3. 创建虚拟环境(可选)

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

    • CPU版本
      pip install torch torchvision torchaudio  
      
    • GPU版本(需CUDA支持)
      访问PyTorch官网,根据CUDA版本选择命令,例如CUDA 11.7:
      pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117  
      
  5. 验证安装

    import torch  
    print(torch.__version__)  
    print(torch.cuda.is_available())  # 若安装GPU版本,应返回True  
    
  6. 安装其他依赖(可选)

    pip install numpy pandas matplotlib  
    

说明

  • 若需从源码编译,需额外安装编译工具及依赖(如libtorch-dev),但推荐优先使用pip或conda安装。
  • 安装完成后,可通过pip install --upgrade命令更新PyTorch版本。

0