PyTorch 本地安装指南：全面支持 macOS 、 Linux 和 Windows 系统

PyTorch 本地安装指南：全面支持 macOS 、 Linux 和 Windows 系统

PyTorch 是一个功能强大的深度学习框架，支持高效的计算图和 GPU 加速，广泛应用于人工智能和机器学习项目中。本文从安装前的准备工作开始，详细介绍了如何使用 conda 和 pip 两种主要的包管理工具来安装 PyTorch，同时涵盖了如何在 macOS 上进行特定的安装配置。对于需要从源码构建 PyTorch 的开发者，本文也提供了相应的操作步骤，确保每个步骤都能顺利完成。无论是初学者还是有经验的开发者，都可以通过本指南轻松安装和配置 PyTorch，快速进入深度学习的实践与研究中。

文章目录

• PyTorch 本地安装指南：全面支持 macOS 、 Linux 和 Windows 系统
• • • 一 安装介绍
    • 二 在 macOS 上安装
    • • 1 安装条件
      • 2 安装 Anaconda
      • 3 使用 pip 安装
      • 4 PyTorch 安装命令
    • 三 验证安装
    • 四 从源码构建 PyTorch
    • 五 参考

预备课：

• Python 安装和依赖管理指南：Conda 与 Pip 📦
• 使用 PyCharm 新建 Python 项目详解 🔧

一 安装介绍

根据需求选择配置 PyTorch 本地安装 ，运行安装命令即可。Stable 是经过测试和支持的稳定版本，适合大多数用户。Preview 是最新的构建版本，包含新特性，但可能未经完全测试和支持。安装前需确保满足相关依赖（如 NumPy），具体取决于所用的包管理工具。推荐使用 Anaconda，它会自动安装所有依赖。也可以安装历史版本的 PyTorch。需要注意，LibTorch 仅适用于 C++ 开发者。

当前使用的 Python 版本为 3.10.15 ，PyTorch 版本为 Stablw（2.5.1） ，运行命令行如下：

$ conda install pytorch::pytorch torchvision torchaudio -c pytorch

安装命令详解如下：

• conda install: 这是 conda 命令行工具用来安装软件包的基本命令。conda 是一个跨平台的包管理器，常用于管理 Python 环境和依赖包。
• pytorch::pytorch: 这是指定安装 PyTorch 的主包。pytorch::pytorch 表示安装来自 pytorch 这个渠道（channel）的 PyTorch 包。前缀 pytorch:: 是为了明确指定包的来源渠道（channel），防止与其他来源的包发生冲突。
• torchvision: 这是 PyTorch 生态系统中的一个附加包，提供了计算机视觉相关的工具和数据集。包括图像处理、模型训练和测试等功能。
• torchaudio: 这是 PyTorch 生态系统中的另一个附加包，主要用于音频处理。它提供了音频数据的读取、预处理、转换等功能。
• -c pytorch: 这个选项指定了包的来源渠道。-c 是 conda 命令中用来指定渠道的参数，pytorch 指明从官方的 PyTorch 渠道下载这些包。通常，这意味着下载的包是由 PyTorch 官方维护和发布的，能够保证兼容性和稳定性。

注： 最新版本的 PyTorch 需要 Python 3.9 或更高版本。

二 在 macOS 上安装

PyTorch 可以在 macOS 上安装和使用。使用 PyTorch 时的体验可能会因操作系统和 GPU 能力有所不同，特别是在处理时间上。

1 安装条件

• macOS 版本： PyTorch 支持 macOS 10.15（Catalina）或更高版本。
• Python 版本： 推荐使用 Python 3.9 至 3.12。可以通过 Anaconda、Homebrew 或 Python 官网来安装 Python。
• 包管理工具： 要安装 PyTorch，需使用 pip 或 Anaconda 两种支持的包管理工具之一。

2 安装 Anaconda

可以通过图形化安装程序或命令行安装程序来安装 Anaconda。如果选择命令行安装，按照以下步骤操作：

1）安装 Miniconda3

Miniconda 是 Anaconda 的精简版，访问 Anaconda 官网下载 或 使用以下命令下载并安装：

# 对于 Apple Silicon 芯片（M1/M2），使用以下命令
$ curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh
$ sh Miniconda3-latest-MacOSX-arm64.sh
# 然后按照提示完成安装

2）使用 conda 来管理环境

安装完成后，可使用 conda 来管理 Python 环境。

3 使用 pip 安装

如果通过 Homebrew 或 Python 官网安装了 Python，pip 会自动随 Python 一起安装。如果安装的是 Python 3.x，需使用 pip3 命令进行安装。

4 PyTorch 安装命令

1） 使用 Anaconda 安装

$ conda install pytorch::pytorch torchvision torchaudio -c pytorch

2）使用 pip 安装

根据 Python 版本，运行以下命令：

# 如果使用的是 Python 3.x，运行：
$ pip3 install torch torchvision

三 验证安装

为了确保 PyTorch 安装成功，可以运行以下代码验证安装：

import torch
x = torch.rand(5, 3)
print(x)

如果安装正确，输出类似于：

tensor([[0.9864, 0.9220, 0.3461],
        [0.5508, 0.5395, 0.3264],
        [0.1688, 0.8139, 0.9784],
        [0.4732, 0.2000, 0.6875],
        [0.9531, 0.9979, 0.6321]])

四 从源码构建 PyTorch

对于大多数用户来说，从预编译的二进制文件安装 PyTorch 是最便捷的方式。然而，如果需要测试最新的 PyTorch 代码或进行核心开发，也可以选择从源码编译安装 PyTorch。

• 可选： 安装 Anaconda。按照 GitHub上的步骤 安装。
• 按照页面中的说明从源码编译 PyTorch。

五 参考

[1] PyTorch 官方文档

[2] PyTorch 本地安装

[3] PyTorch 中文文档