Open tomaz-suller opened 1 month ago
I suspect some incompatibility due to versions of system or other Python packages could be to blame, since I'm running EndeavourOS (rolling release, Arch-based) with Python 3.12.3.
I tried downgrading the onnx
to 1.14.1 but I got a build error from absl
complaining my compiler didn't support C++14 (which is weird since it should but I just gave up then).
Just checked and also in Google Colab I get the same error following the same steps I mentioned, but running on CPU and in Python 3.10.12
@tomaz-suller what version of ONNX are you using? If you haven't already, could you try with onnx==1.15.0? Also, what version of onnxruntime-training are you using?
It does work with onnx==1.15.0
in Colab. I'm using onnx-training-cpu==1.17.3
Edit: locally, I get the ABSL build error about C++14 I mentioned when trying to downgrade, but then the issue isn't with ONNX anymore.
This issue has been automatically marked as stale due to inactivity and will be closed in 30 days if no further activity occurs. If further support is needed, please provide an update and/or more details.
Describe the issue
Trying to execute the example notebook provided in
on_device_training/desktop/python/mnist.ipynb
results in an error about IR version incompatibility, stating the optimiser only supports version <=9 while the generated artifacts use version 10.To reproduce
(initially added them to
requirements.txt
, then installed one-by-one after eachImportError
to check if that wasn't the problem)Execute first cell of the section
which should raise the following error
Urgency
I need to develop on top of this for a project due next month.
ONNX Runtime Installation
Released Package
ONNX Runtime Version or Commit ID
1.17.3
PyTorch Version
2.3.0+cu121
Execution Provider
ROCm
Execution Provider Library Version
ROCm 6.0.2