From 6ffe9284da0d812413a1cc800e7c77bc7988a9da Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Thu, 13 Jun 2024 19:51:35 -0700 Subject: [PATCH 1/2] Add _10 suffix to NVINFER_LIBRARY and NVINFER_PLUGIN_LIBRARY for TRT 10 --- CMakeLists.txt | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 799e5ce..f7925c6 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -216,8 +216,9 @@ 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) +# FIXME: TRT 10 appears to have differently named libraries with "_10" suffix +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 From d454ddf702e0a69b88f0e0236699b5bbb91e7b87 Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Fri, 14 Jun 2024 16:13:39 -0700 Subject: [PATCH 2/2] Add link to TRT 10 Windows release notes --- CMakeLists.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index f7925c6..62fa538 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -216,7 +216,8 @@ FOREACH(p ${TRITON_TENSORRT_LIB_PATHS}) set(TRITON_TENSORRT_LDFLAGS ${TRITON_TENSORRT_LDFLAGS} "-L${p}") ENDFOREACH(p) -# FIXME: TRT 10 appears to have differently named libraries with "_10" suffix +# 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(