Unable to install onnxsim in jetson containers on ORIN AGX

dusty-nv / jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

MIT License

1.89k stars 416 forks source link

Unable to install onnxsim in jetson containers on ORIN AGX #475

Open GrauleM opened 2 months ago

GrauleM commented 2 months ago

Thanks for the great work on setting up these containers!

Unfortunately, I am unable to install onnxsim on my Jetson Orin AGX in the l4t-pytorch containers.

My setup: Jetson AGX Orin 64GB Developer Kit Jetpack Version: 6.0-b52 (confirmed with sudo apt show nvidia-jetpack -a)

Error: Installing onnxsim fails; it ends up in an infinite loop when building the wheels for setup.py

dusty-nv commented 2 months ago

@GrauleM IIRC I had also faced issue in the past with getting onnxsim to install, and ended up just running the model simplification on x86, and then copied the resulting ONNX to Jetson

GrauleM commented 2 months ago

Ok, thanks for the fast response!