如果你正在使用 macOS 操作系统,以下是如何安装 PyTorch 的详细步骤。

系统要求

在开始之前,请确保你的 macOS 符合以下要求:

  • macOS 10.14 或更高版本
  • 系统内存至少为 8GB(推荐 16GB 或更高)

安装步骤

  1. 更新 macOS
    确保你的 macOS 已经更新到最新版本,以便支持 PyTorch 的最新版本。

  2. 安装 Python
    PyTorch 需要 Python 3.6 或更高版本。你可以从 Python 官网 下载并安装。

  3. 安装 Homebrew
    Homebrew 是一个包管理器,可以让你轻松安装软件包。打开终端并运行以下命令来安装 Homebrew:

    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
    
  4. 安装 PyTorch
    使用 Homebrew 安装 PyTorch。打开终端并运行以下命令:

    brew install pytorch
    

    你也可以使用以下命令来安装具有 GPU 支持的 PyTorch:

    brew install pytorch-cuda
    
  5. 验证安装
    打开 Python 终端并运行以下命令来验证 PyTorch 是否已正确安装:

    import torch
    print(torch.__version__)
    

    如果 PyTorch 已正确安装,它将打印出安装的版本号。

扩展阅读

想要了解更多关于 PyTorch 的信息,可以访问我们的 PyTorch 教程页面

图片

macOS PyTorch Installation