Skip to content

Commit fabdacc

Browse files
committed
fix trt net build
1 parent 739e6b7 commit fabdacc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

csrc/mmdeploy/net/trt/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,6 @@ target_include_directories(${PROJECT_NAME} PRIVATE
1010
target_include_directories(${PROJECT_NAME} PRIVATE ${CUDNN_DIR}/include)
1111
target_include_directories(${PROJECT_NAME} PRIVATE ${CUDA_TOOLKIT_ROOT_DIR}/include)
1212
target_link_libraries(${PROJECT_NAME} PRIVATE mmdeploy_tensorrt_ops_obj)
13-
target_link_libraries(${PROJECT_NAME} PUBLIC ${TENSORRT_LIBS} cudnn)
13+
target_link_libraries(${PROJECT_NAME} PUBLIC ${TENSORRT_LIBS} cudnn cuda)
1414

1515
add_library(mmdeploy::trt_net ALIAS ${PROJECT_NAME})

0 commit comments

Comments
 (0)