python - 为什么我不能通过 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