Created
December 2, 2014 22:37
-
-
Save sio2boss/55d116e1f8bf0e73f0d6 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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