Try to run ssd_mobilenet_v1_trt model but crashed

I optimized the ssd_mobilenet_v1 model to ssd_mobilenet_v1_trt model by follow this link https://github.com/NVIDIA-AI-IOT/tf_trt_models.

But when I run the sample code in Juston Nano’s jupyter notebook, it told me that kernel has been crashed. Why?

Hi,

You can check this topic for model ssd_mobilenet_v1 to TensorRT sample:
https://github.com/AastaNV/TRT_object_detection

Thanks.

Thank you.

I’ll try this repo.

And there are the update of my issue. I try to run the origin code in Terminal, there are the error message.

Default Logger The engine plan file is incompatible with this version of TensorRT, expecting 5.0.6.3 got 5.0.2.6, please rebuild

It told me that the version of TensorRT was incompatible, acturally I optimized the model in PC which has GTX 1066 TensorRT 5.0.2.6, and the Jeston Nano’s TensorRT version is 5.0.6.3.

How can I install the TensorRT 5.0.6.3 in my PC?

Hi,

I try to follow the repo but stopped in the environment, this repo use pycuda, but when I try to install the pycuda by pip3, the wheel told me that cuda.h: No such file or directory.

In file included from src/cpp/cuda.cpp:1:0:
      src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory
      #include <cuda.h>
      compilation terminated.
      error: command 'aarch64-linux-gnu-gcc' failed with exit status 1

I found the solution in here https://wiki.tiker.net/PyCuda/Installation/Linux/Ubuntu

export CUDA_ROOT=/usr/local/cuda
export PATH=$PATH:$CUDA_ROOT/bin

Thank you! I succeeded!

Good to hear this : )