diff --git a/csrc/mmdeploy/net/trt/CMakeLists.txt b/csrc/mmdeploy/net/trt/CMakeLists.txt index 5de364162..bc49d0b17 100644 --- a/csrc/mmdeploy/net/trt/CMakeLists.txt +++ b/csrc/mmdeploy/net/trt/CMakeLists.txt @@ -10,6 +10,6 @@ target_include_directories(${PROJECT_NAME} PRIVATE target_include_directories(${PROJECT_NAME} PRIVATE ${CUDNN_DIR}/include) target_include_directories(${PROJECT_NAME} PRIVATE ${CUDA_TOOLKIT_ROOT_DIR}/include) target_link_libraries(${PROJECT_NAME} PRIVATE mmdeploy_tensorrt_ops_obj) -target_link_libraries(${PROJECT_NAME} PUBLIC ${TENSORRT_LIBS} cudnn) +target_link_libraries(${PROJECT_NAME} PUBLIC ${TENSORRT_LIBS} cudnn cuda) add_library(mmdeploy::trt_net ALIAS ${PROJECT_NAME})