mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
41 lines
1.3 KiB
CMake
41 lines
1.3 KiB
CMake
cmake_minimum_required(VERSION 3.20)
|
|
|
|
project(Inceptionv3
|
|
LANGUAGES CXX CUDA
|
|
)
|
|
|
|
add_library(inception_v3_lib
|
|
inception_modules.cpp
|
|
inception_utils.cpp
|
|
)
|
|
|
|
find_library(CUDANet_LIBRARY NAMES CUDANet HINTS ${CMAKE_CURRENT_SOURCE_DIR}/../../build)
|
|
find_package(OpenCV REQUIRED COMPONENTS core imgcodecs imgproc)
|
|
find_package(CUDAToolkit REQUIRED)
|
|
|
|
set (CUDANet_INCLUDE_DIRS
|
|
${CMAKE_CURRENT_SOURCE_DIR}/../../include
|
|
${CMAKE_CURRENT_SOURCE_DIR}/../../include/utils
|
|
${CMAKE_CURRENT_SOURCE_DIR}/../../include/kernels
|
|
${CMAKE_CURRENT_SOURCE_DIR}/../../include/layers
|
|
${CMAKE_CURRENT_SOURCE_DIR}/../../include/model
|
|
)
|
|
|
|
target_include_directories(
|
|
inception_v3_lib PRIVATE ${CMAKE_CURRENT_SOURCE_DIR} ${CUDANet_INCLUDE_DIRS} ${CUDAToolkit_INCLUDE_DIRS})
|
|
|
|
target_link_libraries(inception_v3_lib PRIVATE ${CUDANet_LIBRARY} ${OpenCV_LIBS} CUDA::cudart)
|
|
|
|
set_property(TARGET inception_v3_lib PROPERTY CXX_STANDARD 20)
|
|
|
|
add_executable(inception_v3
|
|
inception_v3.cpp
|
|
)
|
|
|
|
target_include_directories(inception_v3 PRIVATE ${CMAKE_CURRENT_SOURCE_DIR} ${OpenCV_INCLUDE_DIRS} ${CUDANet_INCLUDE_DIRS} ${CUDAToolkit_INCLUDE_DIRS})
|
|
|
|
target_link_libraries(inception_v3 PRIVATE inception_v3_lib ${CUDANet_LIBRARY})
|
|
|
|
set_property(TARGET inception_v3 PROPERTY CXX_STANDARD 20)
|
|
|
|
add_subdirectory(tests) |