Dont ignore cmakelists

This commit is contained in:
2024-05-20 12:31:33 +02:00
parent ba4bee90ad
commit 94df063dcd
2 changed files with 26 additions and 1 deletions

1
.gitignore vendored
View File

@@ -45,4 +45,3 @@ __pycache__
# Test files
*.jpg
*.txt

View File

@@ -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)