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-gnu
and TensorRT libraries are located at/usr/lib/x86_64-linux-gnu
, so using-DTENSORRT_ROOT
will 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_MAJOR
is defined atNvInferVersion.h
instead ofNvInfer.h
, so add compatibility fix.