From 6c099b71935d3e181b7ff06ba6b2eaf4ba750788 Mon Sep 17 00:00:00 2001 From: Francesco Petrini Date: Tue, 27 Aug 2024 09:35:53 -0700 Subject: [PATCH] build: RHEL8 TensorRT Backend (#98) * Add hints --- CMakeLists.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 847cdd1..7a70a85 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -247,8 +247,8 @@ ENDFOREACH(p) # 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) +find_library(NVINFER_LIBRARY NAMES nvinfer nvinfer_10 PATHS "/usr/local/cuda/targets/x86_64-linux/lib") +find_library(NVINFER_PLUGIN_LIBRARY NAMES nvinfer_plugin nvinfer_plugin_10 PATHS "/usr/local/cuda/targets/x86_64-linux/lib") target_link_libraries( triton-tensorrt-backend PRIVATE