yolov5_gpu_optimization
yolov5_gpu_optimization copied to clipboard
Cannot run on NVIDIA Jetson
Hello,
It seems that onnx can only be installed in version 1.6.0 for Jetson Nano.
And when I run the following to convert the model into onnx:
python export.py --weights yolov5s.pt --include onnx --simplify --dynamic
The following error is there:
onnx.onnx_cpp2py_export.checker.ValidationError: Your model ir_version is higher than the checker's
After further researching online, it seems that to get past this error, the latest version of onnx need to be installed. However, if I try:
pip3 install onnx
The installation fails.
Hope anyone can help.
Thank you.
@lakshanthad You don't need to run the export script on Jetson NANO. You can run export on host machine (even without GPU) and run inference on NANO
Okay, thank you. I will try it.
But is there any way to run the export script on Jetson Nano?