加入我们参加旧金山 PyTorch Conference,10 月 22-23 日。CFP 现已开放!了解更多。

PyTorch logo
开始使用

选择您的路径:本地安装 PyTorch 或在支持的云平台上即时启动

开始使用

博客

关注 PyTorch 基金会最新新闻和技术主题

阅读更多

PyTorch 2.6

包含对 torch.compile 的 Python 3.13 支持、多个 AOTInductor 增强、X86 CPU 上的 FP16 支持等功能。

了解更多

会员资格可用

成为 PyTorch 基金会的核心部分,共同构建和塑造人工智能的未来。

加入

关键特性与能力

查看所有功能
生产就绪

使用 TorchScript 在急切模式和图模式之间无缝切换,并通过 TorchServe 加速生产路径。

分布式训练

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

生态系统强大

丰富的工具和库生态系统扩展了 PyTorch,支持计算机视觉、NLP 等领域的发展。

云支持

PyTorch 在主流云平台上得到了良好的支持,提供了无摩擦的开发和易于扩展的功能。

安装 PyTorch

选择您的偏好并运行安装命令。稳定版代表目前测试和受支持的 PyTorch 最新版本,这应该适合大多数用户。如果您想使用每晚生成的最新、未完全测试和受支持的构建,请预览。请确保您已满足以下先决条件(例如,numpy),具体取决于您的包管理器。您还可以安装 PyTorch 的早期版本。请注意,LibTorch 仅适用于 C++。

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

PyTorch 构建
您的操作系统
软件包
语言
计算平台
运行此命令:
PyTorch 构建
稳定版(1.13.0)
预览(夜间版)
您的操作系统
Linux
Mac
Windows
软件包
Conda
Pip
LibTorch
语言
Python
C++ / Java
计算平台
CUDA 11.8
CUDA 12.1
CUDA 12.4
ROCm 5.2
CPU
运行此命令:
conda install pytorch torchvision -c pytorch

PyTorch 的先前版本

快速入门与云合作伙伴

通过流行的云平台和机器学习服务快速使用 PyTorch

公司和大学使用 PyTorch

通过 PyTorch、TorchServe 和 AWS Inferentia 降低推理成本 71%,并推动扩展。

了解更多

推进自然语言处理和多任务学习的艺术水平。

了解更多

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

了解更多

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源