关键特性与能力
查看所有功能生产就绪
使用 TorchScript 在急切模式和图模式之间无缝切换,并通过 TorchServe 加速生产路径。
分布式训练
通过 torch.distributed 后端,实现了在研究和生产中的可扩展分布式训练和性能优化。
生态系统强大
丰富的工具和库生态系统扩展了 PyTorch,支持计算机视觉、NLP 等领域的发展。
云支持
PyTorch 在主流云平台上得到了良好的支持,提供了无摩擦的开发和易于扩展的功能。
安装 PyTorch
选择您的偏好并运行安装命令。稳定版代表目前测试和受支持的 PyTorch 最新版本,这应该适合大多数用户。如果您想使用每晚生成的最新、未完全测试和受支持的构建,请预览。请确保您已满足以下先决条件(例如,numpy),具体取决于您的包管理器。您还可以安装 PyTorch 的早期版本。请注意,LibTorch 仅适用于 C++。
注意:最新版本的 PyTorch 需要 Python 3.9 或更高版本。
PyTorch 的先前版本
探索丰富的库、工具等生态系统,以支持开发。
社区
加入 PyTorch 开发者社区,贡献、学习并获得问题的解答。