Skip to content

Commit 6364bf6

Browse files
committedSep 8, 2022
fix tensorrt cmakelist.txt
1 parent 12c34e1 commit 6364bf6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed
 

‎tensorrt/CMakeLists.txt

+2-1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@ link_directories(${PROJECT_SOURCE_DIR}/build)
1515
find_package(CUDA REQUIRED)
1616
find_package(OpenCV REQUIRED)
1717

18+
cuda_add_library(kernels STATIC kernels.cu)
19+
1820
add_executable(segment segment.cpp trt_dep.cpp)
1921
target_include_directories(
2022
segment PUBLIC ${CUDA_INCLUDE_DIRS} ${CUDNN_INCLUDE_DIRS} ${OpenCV_INCLUDE_DIRS})
@@ -23,4 +25,3 @@ target_link_libraries(
2325
${CUDA_LIBRARIES}
2426
${OpenCV_LIBRARIES})
2527

26-
cuda_add_library(kernels STATIC kernels.cu)

0 commit comments

Comments
 (0)