From b377dcd94ea63d80f4762c1f454d7ad9ad99bbeb Mon Sep 17 00:00:00 2001
From: fpetrini15 <francescogpetrini@gmail.com>
Date: Wed, 14 Aug 2024 16:01:09 -0700
Subject: [PATCH] Add hints

---
 CMakeLists.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 847cdd1..0a2e8e2 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 HINTS "/usr/local/cuda/targets/x86_64-linux/lib")
+find_library(NVINFER_PLUGIN_LIBRARY NAMES nvinfer_plugin nvinfer_plugin_10 HINTS "/usr/local/cuda/targets/x86_64-linux/lib")
 target_link_libraries(
   triton-tensorrt-backend
   PRIVATE