安装 PyTorch 在 macOS 上

在 macOS 上安装 PyTorch 是一个相对简单的过程。以下是一些步骤和提示，帮助您顺利完成安装。

系统要求

在开始之前，请确保您的 macOS 符合以下要求：

• macOS 10.13 或更高版本
• Python 3.6 或更高版本

安装步骤

1. 安装 Python：如果您的系统尚未安装 Python，请先安装 Python 3。您可以从 Python 官网 下载并安装。

2. 打开终端：按下 Command + Space 打开 Spotlight，输入 Terminal 并打开。

3. 更新 pip 和 setuptools：在终端中运行以下命令：

   python3 -m pip install --upgrade pip setuptools
   

4. 安装 PyTorch：转到 PyTorch 官方网站，选择适合您需求的 PyTorch 版本和 macOS 版本。根据页面上的指示，运行相应的命令进行安装。

   例如，如果您想安装 GPU 版本的 PyTorch，运行以下命令：

   pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
   

5. 验证安装：在终端中运行以下命令，以验证 PyTorch 是否已正确安装：

   python3 -c "import torch; print(torch.__version__)"
   

常见问题

以下是一些在安装 PyTorch 时可能会遇到的问题及其解决方案：

• 问题：安装过程中出现错误。

  • 解决方案：检查您是否遵循了正确的步骤，并确保您的系统满足所有要求。

• 问题：安装完成后，无法运行 PyTorch。

  • 解决方案：确保您已将 Python 添加到您的环境变量中。

• 问题：在尝试运行 PyTorch 时遇到 CUDA 错误。

  • 解决方案：确保您的 CUDA 驱动程序与 PyTorch 版本兼容。

更多资源

如果您需要更多帮助或有其他问题，请访问我们的 社区论坛，在那里您可以找到更多资源和讨论。

PyTorch Logo