CMakeLists.txt

cmake_minimum_required(VERSION 3.0)

project(darknet)

set(USE_GPU 1)

add_definitions(-DDEBUG)

if (USE_GPU)
MESSAGE(WARNING "Using GPU")
add_definitions(-DGPU -DCUDN)
SET(CUDA_64_BIT_DEVICE_CODE ON)
LIST(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_30,code=sm_30)
LIST(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_35,code=sm_35)
LIST(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_50,code=[sm_50,compute_50])
LIST(APPEND CUDA_NVCC_FLAGS -gencode arch=compute_52,code=[sm_52,compute_52])

# packages
find_package(CUDA REQUIRED)

else()
    MESSAGE(WARNING "NOT using GPU")
endif()

set(CMAKE_CXX_STANDARD 11)
#set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/../lib)
#set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/../lib)
#set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/../bin)
#set(DARKNET_LIB ${CMAKE_LIBRARY_OUTPUT_DIRECTORY})

set(INCLUDE_DIR include/
        /usr/local/include
        /usr/local/cuda/include)

include_directories(${INCLUDE_DIR})


link_directories(/usr/local/lib
        /usr/local/cuda/lib64)

FILE(GLOB SOURCE_FILES "src/*.c" "src/*.cpp" "src/*.h")
FILE(GLOB CUDA_SOURCE_FILES "src/*.cu")

#add_library(darknet SHARED ${SOURCE_FILES})
if(USE_GPU)
    #cuda_add_executable(darknet ${SOURCE_FILES} ${CUDA_SOURCE_FILES})
    cuda_add_library(darknet  ${SOURCE_FILES} ${CUDA_SOURCE_FILES})

    set(CUDA_LIBS cuda cudart cublas curand cudnn)
else()
    #add_executable(darknet ${SOURCE_FILES})
    add_library(darknet  ${SOURCE_FILES})
endif()

LIST(APPEND CMAKE_CXX_FLAGS "-Wall -Wfatal-errors")


target_link_libraries(darknet
        m
        pthread
        ${CUDA_LIBS}
        )