Pytorch使用手册—（可选）将模型从 PyTorch 导出到 ONNX 并使用 ONNX Runtime 运行（专题三十五）

说明
截至 PyTorch 2.1，ONNX 导出器有两个版本。
torch.onnx.dynamo_export 是最新的（仍处于测试阶段）导出器，基于 PyTorch 2.0 发布的 TorchDynamo 技术。
torch.onnx.export 基于 TorchScript 后端，自 PyTorch 1.2.0 以来一直可用。

在本教程中，我们将介绍如何使用 TorchScript 的 torch.onnx.export ONNX 导出器，将 PyTorch 中定义的模型转换为 ONNX 格式。

导出的模型将通过 ONNX Runtime 执行。ONNX Runtime 是一个专注于性能的 ONNX 模型引擎，能够在多个平台和硬件（Windows、Linux、Mac 以及 CPU 和 GPU）上高效地进行推理。ONNX Runtime 已被证明可以显著提高多种模型的性能，具体说明可以参考这里。

在本教程中，您需要安装 ONNX 和 ONNX Runtime。您可以通过以下方式获取 ONNX 和 ONNX Runtime 的二进制版本：

%