LiMR_cpp/CMakeLists.txt at main · ShowayLiao/LiMR_cpp · 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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
cmake_minimum_required(VERSION 3.10)

project(limr LANGUAGES CUDA CXX)

# define options
option(USE_DIRECT_BLOB "Enable direct blob access" ON)

add_definitions(-DUSE_DIRECT_BLOB=${USE_DIRECT_BLOB})

# Modify to your path
set(OpenCV_DIR "C:/opencv_s/build/install")
set(TRT_DIR "C:/tensorRT/TensorRT-10.12.0.36.Windows.win10.cuda-11.8/TensorRT-10.12.0.36")
set(CMAKE_CUDA_ARCHITECTURES "86")

add_definitions(-std=c++11)
add_definitions(-DAPI_EXPORTS)
set(CMAKE_CXX_STANDARD 11)
set(CMAKE_BUILD_TYPE Debug)

# setup CUDA
find_package(CUDA REQUIRED)
message(STATUS "libraries: ${CUDA_LIBRARIES}")
message(STATUS "include path: ${CUDA_INCLUDE_DIRS}")
include_directories(${CUDA_INCLUDE_DIRS})

enable_language(CUDA)

include_directories(${PROJECT_SOURCE_DIR}/include)

# TensorRT
# find_package(TRT_DIR)
include_directories(${TRT_DIR}/include)
link_directories(${TRT_DIR}/lib)
include_directories(${TRT_DIR}/samples/common)


# OpenCV
find_package(OpenCV)
include_directories(${OpenCV_INCLUDE_DIRS})

# torch
set(Torch_DIR "C:/libtorch/libtorch-win-shared-with-deps-debug-1.13.0+cu116/libtorch/share/cmake/Torch")
find_package(Torch REQUIRED)
include_directories(${Torch_INCLUDE_DIRS})
include_directories(C:/libtorch/libtorch-win-shared-with-deps-debug-1.13.0+cu116/libtorch/include/torch/csrc/api/include)
include_directories(C:/libtorch/libtorch-win-shared-with-deps-debug-1.13.0+cu116/libtorch/include)
link_directories(C:/libtorch/libtorch-win-shared-with-deps-debug-1.13.0+cu116/libtorch/lib)


file(GLOB_RECURSE SRCS ${PROJECT_SOURCE_DIR}/src/*.cpp)
file(GLOB_RECURSE SRCS_CU ${PROJECT_SOURCE_DIR}/src/*.cu)
list(APPEND SRCS ${SRCS_CU})
add_executable(limr ${PROJECT_SOURCE_DIR}/src/main.cpp ${SRCS})


target_link_libraries(limr nvinfer_10)
target_link_libraries(limr cudart)
target_link_libraries(limr nvonnxparser_10)
# target_link_libraries(limr cudaimgproc cudaarithm)
target_link_libraries(limr ${OpenCV_LIBS})
# target_link_libraries(limr ${Torch_LIBS})
target_link_libraries(limr torch)