Skip to content

Instantly share code, notes, and snippets.

@sio2boss
Created December 2, 2014 22:37
Show Gist options
  • Save sio2boss/55d116e1f8bf0e73f0d6 to your computer and use it in GitHub Desktop.
Save sio2boss/55d116e1f8bf0e73f0d6 to your computer and use it in GitHub Desktop.
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 3cb7d58..dc70e8d 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -4,6 +4,7 @@ project( Caffe )
### Build Options ##########################################################################
option(CPU_ONLY "Build Caffe without GPU support" OFF)
+option(CUDNN "Build with cuDNN support" ON)
option(BUILD_PYTHON "Build Python wrapper" OFF)
option(BUILD_MATLAB "Build Matlab wrapper" OFF)
option(BUILD_EXAMPLES "Build examples" ON)
@@ -32,6 +33,8 @@ set(CMAKE_CXX_FLAGS_RELEASE ${CMAKE_CXX_FLAGS_RELEASE}) # set release flags
# Global Definitions
if(CPU_ONLY)
add_definitions(-DCPU_ONLY)
+elseif (CUDNN)
+ add_definitions(-DUSE_CUDNN)
endif()
# Include Directories
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment