tensorrt_smoke/CMakeLists.txt at main · storrrrrrrrm/tensorrt_smoke · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
cmake_minimum_required(VERSION 2.8.0)
project(smoke_trt)
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_BUILD_TYPE Debug)

find_package(CUDA REQUIRED)
find_package(OpenCV REQUIRED)
find_library(NVINFER NAMES nvinfer)
find_library(NVINFER_PLUGIN NAMES nvinfer_plugin)
find_library(NVONNXPARSER nvonnxparser)


include_directories(
  ${CUDA_INCLUDE_DIRS}
  # ~/TensorRT-8.0.3.4/include
  plugins/modulated_deform_conv
)

cuda_add_library(trt_modulated_deform_conv_plugin SHARED
  plugins/modulated_deform_conv/trt_modulated_deform_conv_kernel.cu
  plugins/modulated_deform_conv/trt_modulated_deform_conv.cpp
)

add_executable(${PROJECT_NAME}
  smoke.cpp
  main.cpp
)

target_link_libraries(${PROJECT_NAME}
  ${NVINFER}
  ${NVONNXPARSER}
  ${NVINFER_PLUGIN}
  ${CUDA_LIBRARIES}
  ${OpenCV_LIBRARIES}
  trt_modulated_deform_conv_plugin
)