added cuda libraries to cmake link_libraries of TensorRT, required fo…

…r TensorRT 7.2 on linux
smistad · Dec 1, 2020 · b8a12fc · b8a12fc
1 parent 4a6f5f0
commit b8a12fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)