fix: Add _10 version suffix to NVINFER_LIBRARY and NVINFER_PLUGIN_LIB…

…RARY for TRT 10 Windows Build (#93) (#94)
triton-inference-server · Jun 17, 2024 · 9c5effd · 9c5effd
1 parent fd867b7
commit 9c5effd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -212,8 +212,10 @@ FOREACH(p ${TRITON_TENSORRT_LIB_PATHS})
     set(TRITON_TENSORRT_LDFLAGS ${TRITON_TENSORRT_LDFLAGS} "-L${p}")
 ENDFOREACH(p)
 
-find_library(NVINFER_LIBRARY NAMES nvinfer)
-find_library(NVINFER_PLUGIN_LIBRARY NAMES nvinfer_plugin)
+# NOTE: TRT 10 for Windows added the version suffix to the library names. See the release notes:
+# https://docs.nvidia.com/deeplearning/tensorrt/release-notes/index.html#tensorrt-10
+find_library(NVINFER_LIBRARY NAMES nvinfer nvinfer_10)
+find_library(NVINFER_PLUGIN_LIBRARY NAMES nvinfer_plugin nvinfer_plugin_10)
 target_link_libraries(
   triton-tensorrt-backend
   PRIVATE