以下是 PyTorch 的安装步骤，确保您按照以下指南进行操作。

系统要求

  • Python 3.6+
  • CUDA (可选，用于 GPU 加速)

安装步骤

  1. 安装 Python：首先，请确保您已安装 Python 3.6 或更高版本。您可以从 Python 官网 下载并安装。

  2. 安装 PyTorch：

    • CPU 版本：如果您没有 GPU，可以使用 CPU 版本的 PyTorch。在终端中运行以下命令：

      pip install torch torchvision torchaudio

    • GPU 版本：如果您有 GPU，您需要安装 GPU 版本的 PyTorch。请访问 PyTorch 官网，选择适合您 CUDA 版本的 PyTorch，并按照指南进行安装。

  3. 验证安装：安装完成后，在终端中运行以下命令来验证 PyTorch 是否已正确安装：

    import torch
print(torch.__version__)

    如果一切正常，您将看到 PyTorch 的版本号。

其他资源

如果您需要更多帮助，请访问以下资源：

图片示例

以下是 PyTorch 的一个示例：

PyTorch Logo