SSD Lite Mobilenet V2 returns shifted bounding boxes after conversion with TensorRT 8.0

[AastaLLL]

Hi,

For “No CMAKE_CUDA_COMPILER could be found” error:

Please add nvcc to the PATH environment variable.
For example:

$ echo 'export PATH=/usr/local/cuda-10.2/bin:$PATH' >> ~/.bashrc
$ echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
$ source ~/.bashrc

For “cub/cub.cuh” error:

Please try following patch to see if it works:

diff --git a/plugin/CMakeLists.txt b/plugin/CMakeLists.txt
index 562681f..656cf9e 100644
--- a/plugin/CMakeLists.txt
+++ b/plugin/CMakeLists.txt
@@ -77,7 +77,7 @@ if(BERT_GENCODES)
     include_directories(bertQKVToContextPlugin/fused_multihead_attention/include bertQKVToContextPlugin/fused_multihead_attention_v2/include)
 endif()

-include_directories(common common/kernels ${CMAKE_SOURCE_DIR}/third_party)
+include_directories(common common/kernels ${CMAKE_SOURCE_DIR}/third_party ${CMAKE_SOURCE_DIR}/third_party/cub)

 foreach(PLUGIN_ITER ${PLUGIN_LISTS})
     include_directories(${PLUGIN_ITER})

Thanks.

[sdelyon]

Hi,

Thank you very much. Both of the errors I had could be fixed with your previous message so I could build the new patched plugin which works fine with the bounding boxes correctly placed.