1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
cmake_minimum_required(VERSION 3.1)
project(custom_build_project)
set(CMAKE_CXX_STANDARD 14 CACHE STRING "The C++ standard whose features are requested to build this target.")
# Find torch library
find_package(Torch REQUIRED)
# Main executable
add_executable(Predictor predictor.cpp)
target_include_directories(Predictor PUBLIC ${TORCH_INCLUDE_DIRS})
find_package(Threads REQUIRED)
target_link_libraries(Predictor
-Wl,-s
-Wl,--gc-sections
-Wl,--whole-archive
${TORCH_LIBRARIES}
-Wl,--no-whole-archive
Threads::Threads
${CMAKE_DL_LIBS}
)
|
