为什么我不能在 pytorch 中使用 ONNX 运行时训练？

Question

当我跑步时

from onnxruntime.capi.ort_trainer import ORTTrainer

如https://github.com/microsoft/onnxruntime/#training-start 所述，我收到此错误：

ModuleNotFoundError: No module named 'onnxruntime.capi.ort_trainer'

我能做些什么来解决这个问题？ 我通过 pip 安装了 onnxruntime，但我什至在 [python path]/site_packages/onnx-runtime/capi 中找不到“ort_trainer.py”

Answer 1

您应该从源代码构建 ort 训练。 https://www.onnxruntime.ai/docs/how-to/build.html#training