yolov7 icon indicating copy to clipboard operation
yolov7 copied to clipboard

ERROR while convert to TensorRT from ONNX on Jetson Nano

Open BNTzHax opened this issue 2 years ago • 2 comments

Hello

I have converted successfully Yolov7-Tiny to ONNX via https://colab.research.google.com/github/WongKinYiu/yolov7/blob/main/tools/YOLOv7onnx.ipynb#scrollTo=VaPGM88g-_CE tutorial

But when I try to convert ONNX to TensorRT on Jetson Nano I get this error: "Error Code 2: Internal Error (Assertion status == kSTATUS_SUCCESS failed.)" Here is my Jetson Nano system information: CUDA: 10.2.300 TensorRT: 8.0.1.6 cuDNN: 8.2.1.32

Any help will be appreciate

BNTzHax avatar Jul 28 '22 10:07 BNTzHax

I recently did the same conversion on a GTX1080. I was also getting an error when converting from ONNX to TRT. Error was solved when I installed PyTorch 1.12. Using version 1.10 didn't work.

Please check this issue

GiorgosBetsos avatar Jul 28 '22 11:07 GiorgosBetsos

I recently did the same conversion on a GTX1080. I was also getting an error when converting from ONNX to TRT. Error was solved when I installed PyTorch 1.12. Using version 1.10 didn't work.

Please check this issue

Thank you, i will try that

BNTzHax avatar Aug 02 '22 01:08 BNTzHax

how do you convert the onnx model to trt on a nano?

dlavrantonis avatar Dec 24 '22 13:12 dlavrantonis