欢迎来到 PyTorch 安装指南页面!以下内容将指导您如何安装 PyTorch,并开始使用这个强大的深度学习框架。

系统要求

在安装 PyTorch 之前,请确保您的系统满足以下要求:

  • 操作系统:Linux、macOS 或 Windows
  • Python 版本:3.6 或更高版本

安装步骤

以下是安装 PyTorch 的步骤:

  1. 安装 Python:如果您还没有安装 Python,请从 Python 官网 下载并安装最新版本的 Python。

  2. 安装 PyTorch:打开命令行工具,输入以下命令:

    pip install torch torchvision torchaudio
    

    如果您需要使用 GPU 加速,请使用以下命令:

    pip install torch torchvision torchaudio torchvision-cuda
    
  3. 验证安装:安装完成后,可以通过以下命令验证 PyTorch 是否已正确安装:

    import torch
    print(torch.__version__)
    

    如果输出 PyTorch 版本号,则表示安装成功。

注意事项

  • 在安装过程中,如果遇到任何问题,可以查阅 PyTorch 安装文档 获取帮助。
  • 为了更好地使用 PyTorch,建议您阅读 PyTorch 教程 学习更多基础知识。

示例图片

下面是一张 PyTorch 图标的示例图片:

PyTorch