我们不能运行导入到 pytorch 的 onnx 模型吗？答案

【问题标题】：Can't we run an onnx model imported to pytorch?我们不能运行导入到 pytorch 的 onnx 模型吗？
【发布时间】：2020-03-09 01:25:45
【问题描述】：

我一直在尝试从 onnx 格式导入模型以使用 pytorch。我发现很难找到同样的例子。正如互联网上的大多数资源都在谈论将 pytorch 模型导出到 onnx。

我发现torch.onnx() 只能导出模型，导入方法还没有实现。直接安装 onnx 库，帮助我做 onnx.load("model_name.onnx")。如何将此模型与 pytorch 一起使用？我无法使用 model.to(device="GPU")

将模型移动到 GPU

【问题讨论】：

标签： pytorch onnx

【解决方案1】：

PyTorch 目前不支持导入 onnx 模型。在撰写此答案时，它是open feature request。

虽然不能保证有效，但一个潜在的解决方案是使用由 Microsoft 开发的名为 MMdnn 的工具（不，它不仅限于 Windows！），它支持与各种框架之间的转换。不幸的是，onnx 只能是转换的目标，而不是源。也就是说，您也许能够将您的模型导入另一个框架，然后使用 MMdnn 从该框架转换为 pytorch。显然这并不理想，成功的可能性将取决于其他框架如何使用 onnx，这可能不适合 MMdnn 的工作方式。

【讨论】：

既然现在caffe2和pytorch合并了，有什么办法可以在使用caffe2再在pytorch中使用onnx模型？
我不知道有什么方法可以将 caffe2 模型转换为 pytorch。

【解决方案2】：

有一个名为 onnx2pytorch 的库可以帮助您将 onnx 模型转换为 PyTorch。

【讨论】：