Skip to content
Snippets Groups Projects
Commit 32c24fa8 authored by Jacob Becker's avatar Jacob Becker
Browse files

fix cmake for cudnn support

parent 4f35852e
Branches fix-cudnn
No related tags found
1 merge request!3fix cmake for cudnn support
......@@ -130,12 +130,28 @@ set( headers
set( libs m )
set(CUDNN_FLAGS)
if (USE_CUDNN)
find_library( CUDNN_LIBRARIES
NAMES cudnn libcudnn.so libcudnn.so.5
PATHS ${CUDA_TOOLKIT_ROOT_DIR}/targets/aarch64-linux
${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES lib lib64
)
set(CUDNN_FLAGS -DCUDNN )
list( APPEND libs ${CUDNN_LIBRARIES} )
add_definitions( -DCUDNN )
endif()
if (USE_GPU)
find_package(CUDA QUIET REQUIRED)
include_directories( ${CUDA_TOOLKIT_ROOT_DIR}/include )
add_definitions( -DGPU )
set( CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS}
set( CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} ${CUDNN_FLAGS}
-gencode arch=compute_20,code=[sm_20,sm_21]
-gencode arch=compute_30,code=sm_30
-gencode arch=compute_35,code=sm_35
......@@ -151,18 +167,6 @@ if (USE_GPU)
)
endif()
if (USE_CUDNN)
find_library( CUDNN_LIBRARIES
NAMES cudnn libcudnn.so libcudnn.so.5
PATHS ${CUDA_TOOLKIT_ROOT_DIR}/targets/aarch64-linux
${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES lib lib64
)
list( APPEND libs ${CUDNN_LIBRARIES} )
add_definitions( -DCUDNN )
endif()
if (USE_OPENCV)
find_package( OpenCV 2.4.6 REQUIRED )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment