We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 739e6b7 commit fabdaccCopy full SHA for fabdacc
csrc/mmdeploy/net/trt/CMakeLists.txt
@@ -10,6 +10,6 @@ target_include_directories(${PROJECT_NAME} PRIVATE
10
target_include_directories(${PROJECT_NAME} PRIVATE ${CUDNN_DIR}/include)
11
target_include_directories(${PROJECT_NAME} PRIVATE ${CUDA_TOOLKIT_ROOT_DIR}/include)
12
target_link_libraries(${PROJECT_NAME} PRIVATE mmdeploy_tensorrt_ops_obj)
13
-target_link_libraries(${PROJECT_NAME} PUBLIC ${TENSORRT_LIBS} cudnn)
+target_link_libraries(${PROJECT_NAME} PUBLIC ${TENSORRT_LIBS} cudnn cuda)
14
15
add_library(mmdeploy::trt_net ALIAS ${PROJECT_NAME})
0 commit comments