以下是 PyTorch 在 Linux 系统上的安装步骤,确保您的系统满足以下要求:

  • 操作系统:Linux
  • Python 版本:Python 3.6, 3.7, 或 3.8
  • pip 版本:至少 19.0

安装步骤

  1. 更新系统包列表

    sudo apt-get update
    
  2. 安装 Python 和 pip

    sudo apt-get install python3 python3-pip
    
  3. 安装 PyTorch 打开终端,使用以下命令安装 PyTorch:

    pip3 install torch torchvision torchaudio
    
  4. 验证安装 运行以下 Python 代码来验证 PyTorch 是否已正确安装:

    import torch
    print(torch.__version__)
    

注意事项

  • 如果您需要使用 GPU 加速,请确保安装了 CUDA 和 cuDNN。
  • 您可以访问 PyTorch 官方文档 获取更多详细信息和安装选项。

PyTorch Logo

返回首页