PyTorch 是一个流行的开源机器学习库，适用于深度学习和计算。以下是如何在您的系统上安装 PyTorch 的步骤。

系统要求

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

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

安装步骤

  1. 选择合适的 PyTorch 版本
    访问 PyTorch 官方网站，根据您的操作系统和 Python 版本选择合适的 PyTorch 版本。

  2. 安装 PyTorch
    使用以下命令进行安装：

    pip install torch torchvision torchaudio

    如果您需要 GPU 支持，请使用以下命令：

    pip install torch torchvision torchaudio cpuonly

  3. 验证安装
    打开 Python 解释器并运行以下代码：

    import torch
print(torch.__version__)

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

扩展阅读

如果您想了解更多关于 PyTorch 的信息，可以访问以下链接：

PyTorch Logo