From d32890e7ffefaf2b4fa9056ba485735cbda44494 Mon Sep 17 00:00:00 2001 From: Wang Xinyu Date: Tue, 20 Dec 2022 12:46:24 +0800 Subject: [PATCH] [Bug Fix] fix cmake TRT lib not found bug (#918) fix cmake bug --- FastDeploy.cmake.in | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/FastDeploy.cmake.in b/FastDeploy.cmake.in index 3ad85fb79..44e7eb8a6 100644 --- a/FastDeploy.cmake.in +++ b/FastDeploy.cmake.in @@ -145,7 +145,7 @@ if(WITH_GPU) find_library(TRT_ONNX_LIB nvonnxparser /usr/lib/aarch64-linux-gnu/) find_library(TRT_PLUGIN_LIB nvinfer_plugin /usr/lib/aarch64-linux-gnu/) else() - if(TRT_DIRECTORY) + if(EXISTS ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/) find_library(TRT_INFER_LIB nvinfer ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/lib NO_DEFAULT_PATH) find_library(TRT_ONNX_LIB nvonnxparser ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/lib NO_DEFAULT_PATH) find_library(TRT_PLUGIN_LIB nvinfer_plugin ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tensorrt/lib NO_DEFAULT_PATH) @@ -288,7 +288,6 @@ message(STATUS " ENABLE_VISION : ${ENABLE_VISION}") message(STATUS " ENABLE_TEXT : ${ENABLE_TEXT}") if(WITH_GPU) message(STATUS " CUDA_DIRECTORY : ${CUDA_DIRECTORY}") - message(STATUS " TRT_DIRECTORY : ${TRT_DIRECTORY}") endif() if(OPENCV_DIRECTORY) message(STATUS " OPENCV_DIRECTORY : ${OPENCV_DIRECTORY}")