diff options
-rw-r--r-- | CMakeLists.txt | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 8f9b50e..b4deeed 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -12,10 +12,10 @@ ENDIF() FIND_PACKAGE(Torch REQUIRED) FIND_PACKAGE(CUDA 7.0 REQUIRED) -FIND_PACKAGE(CUDNN 5.1 EXACT QUIET) +FIND_PACKAGE(CUDNN 5 EXACT QUIET) IF(NOT CUDNN_FOUND) CUDNN_INSTALL(5.1 "${Torch_INSTALL_LIB}" "${Torch_INSTALL_INCLUDE}" "") - FIND_PACKAGE(CUDNN 5.1 EXACT REQUIRED) + FIND_PACKAGE(CUDNN 5 EXACT REQUIRED) ENDIF() |