Skip to content

D
dlib
Commit 4fac9804 authored by Daniel Crispell's avatar Daniel Crispell
Browse files
Options

allow user to interactively set CUDNN paths

parent d7b5fc5e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
dlib/CMakeLists.txt
View file @ 4fac9804
...@@ -455,6 +455,9 @@ if (NOT TARGET dlib) ...@@ -455,6 +455,9 @@ if (NOT TARGET dlib)
# linux). # linux).
list(APPEND CUDA_NVCC_FLAGS "-arch=sm_30;-std=c++11;-D__STRICT_ANSI__") list(APPEND CUDA_NVCC_FLAGS "-arch=sm_30;-std=c++11;-D__STRICT_ANSI__")
set(CUDNN_INCLUDE_DIR "" CACHE PATH "Directory in which to look for cudnn include files")
set(CUDNN_LIBRARY_DIR "" CACHE PATH "Directory in which to look for cudnn library")
include(dnn/test_for_cudnn/find_cudnn.txt) include(dnn/test_for_cudnn/find_cudnn.txt)
if (cudnn AND NOT DEFINED cuda_test_compile_worked AND NOT DEFINED cudnn_test_compile_worked) if (cudnn AND NOT DEFINED cuda_test_compile_worked AND NOT DEFINED cudnn_test_compile_worked)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment