Created by: jeng1220
-
Add new search path for TensorRT at tensorrt.cmake. In NVIDIA official docker image, TensorRT headers are located at
/usr/include/x86_64-linux-gnuand TensorRT libraries are located at/usr/lib/x86_64-linux-gnu, so using-DTENSORRT_ROOTwill fail to detect TensorRT. -
Add better debug message. There is no debug/warning message to tell developer that TensorRT is failed to be detected.
-
Fix the bug of detection of TensorRT version. In later version of TensorRT (e.g. v6),
NV_TENSORRT_MAJORis defined atNvInferVersion.hinstead ofNvInfer.h, so add compatibility fix.