search

jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
https://jkjung-avt.github.io/
MIT License
1.74k stars 543 forks source link

python3 onnx_to_tensorrt causes jetson to shut down #508

Closed lpkoh closed 2 years ago

lpkoh commented 2 years ago

I have converted my customer model with 720p resolution yolov4-csp custom model with 3 classes to onnx. However, when I ran the next script to convert this to tensorrt, my jetson agx xavier shuts down after about 10 seconds of the code running. I've retried many times and this always happens.

Is there anyone who has encountered this issue before?

lpkoh commented 2 years ago

Update: this issue does not only happen when I use a 1280 x 720 model with 3 classes (custom), but also with yolov4-csp and yolov4 that I downloaded directly from alexeyab.

lpkoh commented 2 years ago

I am looking to get demo 5 to work. My tensorrt version is 7.1.3, onnx is 1.4.1, pycuda is installed from pip3 and is (2019,1,2). I have run make and I do have a libyolo_layer.so file.

When you say tensorrt requires 6.x+, does this include 7? Or do I need to downgrade.

lpkoh commented 2 years ago

Oh I realized this is due to the nvpmodel setting. Closed.