From 6c4aad8fe01f61f0b0c6bdc0638b280a4e292325 Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Fri, 14 Jun 2024 22:53:39 -0700 Subject: [PATCH] fix: Add _10 version suffix to NVINFER_LIBRARY and NVINFER_PLUGIN_LIBRARY for TRT 10 Windows Build (#93) --- CMakeLists.txt | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index c13034c..7e87e62 100644 --- 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