From b8a12fc057bea0c7b84f376ffe9e2887f7322552 Mon Sep 17 00:00:00 2001 From: Erik Smistad Date: Tue, 1 Dec 2020 16:28:40 +0100 Subject: [PATCH] added cuda libraries to cmake link_libraries of TensorRT, required for TensorRT 7.2 on linux --- .../Algorithms/NeuralNetwork/InferenceEngines/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/source/FAST/Algorithms/NeuralNetwork/InferenceEngines/CMakeLists.txt b/source/FAST/Algorithms/NeuralNetwork/InferenceEngines/CMakeLists.txt index a2f3d14e6..8fc9c3371 100644 --- a/source/FAST/Algorithms/NeuralNetwork/InferenceEngines/CMakeLists.txt +++ b/source/FAST/Algorithms/NeuralNetwork/InferenceEngines/CMakeLists.txt @@ -21,7 +21,7 @@ if(FAST_MODULE_TensorRT) add_library(InferenceEngineTensorRT SHARED TensorRTEngine.hpp TensorRTEngine.cpp) target_include_directories(InferenceEngineTensorRT PRIVATE ${FAST_INCLUDE_DIRS} ${TensorRT_INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS}) - target_link_libraries(InferenceEngineTensorRT FAST ${TensorRT_LIBRARIES}) + target_link_libraries(InferenceEngineTensorRT FAST ${TensorRT_LIBRARIES} ${CUDA_LIBRARIES}) generate_export_header(InferenceEngineTensorRT EXPORT_FILE_NAME ${PROJECT_BINARY_DIR}/TensorRTExport.hpp) fast_add_inference_engine(TensorRT)