主要特性和
功能
查看所有功能
生产就绪
使用 TorchScript 在 eager 模式和图模式之间无缝切换,并使用 TorchServe 加快生产路径。
分布式训练
torch.distributed 后端支持在研究和生产中进行可扩展的分布式训练和性能优化。
强大的生态系统
丰富的工具和库生态系统扩展了 PyTorch,并支持计算机视觉、NLP 等领域的开发。
云支持
PyTorch 在主流云平台上都得到了良好的支持,可提供无缝的开发和轻松的扩展。
安装 PyTorch
选择您的偏好设置并运行安装命令。稳定版表示经过最新测试和支持的 PyTorch 版本。这应该适用于许多用户。如果您想要最新的、未经完全测试和支持的夜间构建版本,可以使用预览版。请确保您已满足以下先决条件(例如,numpy),具体取决于您的软件包管理器。Anaconda 是我们推荐的软件包管理器,因为它会安装所有依赖项。您还可以安装以前版本的 PyTorch。请注意,LibTorch 仅适用于 C++。
注意:最新的 PyTorch 需要 Python 3.8 或更高版本。
以前版本的 PyTorch
探索丰富的库、工具等生态系统,为开发提供支持。
社区
加入 PyTorch 开发者社区,为社区做出贡献、学习并获得问题的解答。