以为 PyTorch 只是一个库？再想想。

这是每位机器学习工程师都应该了解的 PyTorch 生态系统：

• PyTorch – 核心深度学习框架 https://pytorch.org

• LibTorch – 面向生产推理的 C++ API https://pytorch.org/cppdocs

• ExecuTorch – 面向移动端、边缘端和嵌入式设备的端侧 AI https://pytorch.org/executorch

• TorchServe – 面向生产部署的模型服务 https://github.com/pytorch/serve

• TorchAO – 量化与低精度优化 https://github.com/pytorch/ao

• TorchTitan – 大规模训练大型语言模型 https://github.com/pytorch/torchtitan…

• TorchRL – 强化学习库 https://pytorch.org/rl

• TorchVision – 计算机视觉模型与数据集 https://pytorch.org/vision

• TorchAudio – 语音与音频处理 https://pytorch.org/audio

• TorchText – NLP 数据集与预处理 https://pytorch.org/text

• PyTorch Geometric (PyG) – 图神经网络 https://pytorch-geometric.readthedocs.io

• Pyro – 基于 PyTorch 的概率编程 https://pyro.ai

• Lightning – PyTorch 的高级训练框架 https://lightning.ai

掌握 PyTorch 生态系统与掌握 PyTorch 本身同等重要。

PyTorch

来源：https://pytorch.org/ 开始使用：在本地安装 PyTorch，或在支持的云平台上即时启动

摘要：PyTorch 测试通常在导入时生成，因此 CI 失败可能显示与源模板不同的设备/数据类型特定名称。本地调试时，可使用 pytest -k 和 test/run_test.py…（阅读更多）

上周末在旧金山，构建者、研究人员、移动开发者以及 AI 从业者齐聚 ExecuTorch 黑客松，这是一场为期两天的线下活动，聚焦实用且日益重要的主题…（阅读更多）

PyTorch 基金会（Linux 基金会旗下的开源 AI 社区驱动中心）宣布 Shopify 已作为白金会员加入。Shopify 提供关键的互联网基础设施…（阅读更多）

加入 PyTorch 基金会

作为 PyTorch 基金会的成员，您将获得稳定、安全且长期维护的代码库的管理资源。您可以合作进行培训、本地及区域活动、开源开发者工具、学术研究，以及帮助新用户和贡献者获得高效体验的指南。

探索权益 (https://pytorch.org/join)

主要特性与能力

生产就绪

通过 TorchScript 在 eager 模式和图模式之间无缝切换，并借助 TorchServe 加速生产部署。

分布式训练

通过 torch.distributed 后端在研究和生产中实现可扩展的分布式训练与性能优化。

健壮的生态系统

丰富的工具和库生态系统扩展了 PyTorch，支持计算机视觉、自然语言处理等领域的开发。

云支持

PyTorch 在主流云平台上得到良好支持，提供无摩擦的开发体验和轻松扩展。

安装 PyTorch

选择您的偏好并运行安装命令。稳定版代表当前测试和支持最充分的 PyTorch 版本，适合大多数用户。预览版提供最新但未完全测试和支持的每日构建版本。请确保您已满足以下前提条件（例如，numpy），具体取决于您的包管理器。您也可以安装 PyTorch 的历史版本 (https://pytorch.org/get-started/previous-versions)。注意，LibTorch 仅适用于 C++。

注意： 最新的稳定版 PyTorch 需要 Python 3.10 或更高版本。

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

PyTorch 历史版本 (https://pytorch.org/get-started/previous-versions/)

通过云合作伙伴快速开始

通过流行的云平台和机器学习服务快速上手 PyTorch。

生态系统

精选项目

探索丰富的库、工具等资源，支持开发。

Captum

Captum（拉丁语意为“理解”）是一个基于 PyTorch 构建的开源、可扩展的模型可解释性库。

PyTorch Geometric

PyTorch Geometric 是一个用于不规则输入数据（如图、点云和流形）深度学习的库。

skorch

skorch 是 PyTorch 的高级库，提供完整的 scikit-learn 兼容性。

使用 PyTorch 的公司与大学

Amazon Advertising

使用 PyTorch、TorchServe 和 AWS Inferentia 将推理成本降低 71% 并实现扩展。

阅读案例研究 (https://pytorch.org/case-studies/)

Salesforce

推动自然语言处理和多任务学习领域的最前沿技术。

Stanford University

利用 PyTorch 的灵活性高效研究新的算法方法。