vstrt/{trtexec}/CMakeLists.txt: linking cuDNN

AmusementClub · Oct 22, 2023 · 019443f · 019443f
1 parent 14c2622
commit 019443f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vstrt/CMakeLists.txt b/vstrt/CMakeLists.txt
@@ -39,7 +39,7 @@ endif()
 
 if (USE_NVINFER_PLUGIN)
     add_definitions(-DUSE_NVINFER_PLUGIN)
-    find_library(CUDNN_LIB cudnn HINTS ${CUDNN_ROOT_DIR} PATH_SUFFIXES lib)
+    find_library(CUDNN_LIB cudnn HINTS ${CUDNN_ROOT_DIR} PATH_SUFFIXES lib/x64)
     if (USE_NVINFER_PLUGIN_STATIC)
         target_link_libraries(vstrt PRIVATE nvinfer_plugin_static ${CUDNN_LIB} CUDA::cublas)
 

diff --git a/vstrt/trtexec/CMakeLists.txt b/vstrt/trtexec/CMakeLists.txt
@@ -29,7 +29,7 @@ target_link_directories(trtexec PRIVATE
     ${TENSORRT_HOME}/lib
 )
 
-find_library(CUDNN_LIB cudnn HINTS ${CUDNN_ROOT_DIR} PATH_SUFFIXES lib)
+find_library(CUDNN_LIB cudnn HINTS ${CUDNN_ROOT_DIR} PATH_SUFFIXES lib/x64)
 
 target_link_libraries(trtexec PRIVATE
     CUDA::cudart_static