主要特性 &
功能
查看所有特性
生产就绪
使用 TorchScript 在 Eager 模式和图模式之间无缝过渡，并通过 TorchServe 加速生产路径。
分布式训练
torch.distributed 后端支持研究和生产中可扩展的分布式训练和性能优化。
强大的生态系统
丰富的工具和库生态系统扩展了 PyTorch，并支持计算机视觉、自然语言处理等领域的开发。
云支持
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 开发者社区，贡献力量、学习知识并获得解答。
