fix cmake var

pull/219/head
triplemu 6 months ago
parent 5e85b51521
commit 56dc3e2ab7
  1. 14
      csrc/detect/end2end/CMakeLists.txt

@ -25,7 +25,7 @@ message(STATUS "OpenCV Headers: \n${OpenCV_INCLUDE_DIRS}\n")
# TensorRT
set(TensorRT_INCLUDE_DIRS /usr/include/x86_64-linux-gnu)
set(TensorRT_LIBRARIES nvinfer nvinfer_plugin)
list(APPEND TensorRT_LIBRARIES nvinfer nvinfer_plugin)
set(TensorRT_LIB_DIR /usr/lib/x86_64-linux-gnu)
@ -38,25 +38,25 @@ list(APPEND ALL_INCLUDE_DIRS
${OpenCV_INCLUDE_DIRS}
${TensorRT_INCLUDE_DIRS}
include
)
)
list(APPEND ALL_LIBS
${CUDA_LIBRARIES}
${OpenCV_LIBRARIES}
${TensorRT_LIBRARIES}
)
)
message(STATUS "ALL_LIBS: \n${ALL_LIBS}\n")
list(APPEND ALL_LIB_DIRS
${CUDA_LIB_DIR}
${TensorRT_LIB_DIR}
)
message(STATUS "ALL_LIB_DIRS: \n${ALL_LIB_DIRS}\n")
add_executable(${PROJECT_NAME}
main.cpp
include/yolov8.hpp
include/common.hpp
)
)
target_include_directories(${PROJECT_NAME} PUBLIC ${ALL_INCLUDE_DIRS})
target_link_directories(${PROJECT_NAME} PUBLIC ${ALL_LIB_DIRS})
target_link_libraries(${PROJECT_NAME} PRIVATE nvinfer ${ALL_LIBS}})
target_link_libraries(${PROJECT_NAME} PRIVATE ${ALL_LIBS})

Loading…
Cancel
Save