From 94df063dcdaa80044e8302c443576c5fcd262eaf Mon Sep 17 00:00:00 2001 From: LordMathis Date: Mon, 20 May 2024 12:31:33 +0200 Subject: [PATCH] Dont ignore cmakelists --- .gitignore | 1 - examples/inception_v3/CMakeLists.txt | 26 ++++++++++++++++++++++++++ 2 files changed, 26 insertions(+), 1 deletion(-) create mode 100644 examples/inception_v3/CMakeLists.txt diff --git a/.gitignore b/.gitignore index cb75514..9b05bea 100644 --- a/.gitignore +++ b/.gitignore @@ -45,4 +45,3 @@ __pycache__ # Test files *.jpg -*.txt \ No newline at end of file diff --git a/examples/inception_v3/CMakeLists.txt b/examples/inception_v3/CMakeLists.txt new file mode 100644 index 0000000..02d33ed --- /dev/null +++ b/examples/inception_v3/CMakeLists.txt @@ -0,0 +1,26 @@ +cmake_minimum_required(VERSION 3.17) + +project(Inceptionv3 + LANGUAGES CXX CUDA +) + +add_executable(inception_v3 inception_v3.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 PRIVATE ${OpenCV_INCLUDE_DIRS} ${CUDANet_INCLUDE_DIRS} ${CUDAToolkit_INCLUDE_DIRS}) + +target_link_libraries(inception_v3 PRIVATE ${CUDANet_LIBRARY} ${OpenCV_LIBS} CUDA::cudart) + +set_property(TARGET inception_v3 PROPERTY CXX_STANDARD 20) \ No newline at end of file