diff options
author | Soumith Chintala <soumith@gmail.com> | 2017-03-01 15:16:22 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-03-01 15:16:22 +0300 |
commit | 995b582bcb8045f10ffd2995fc692dd13623d4a0 (patch) | |
tree | c4686b0bb904ca91eac5d375a130a4d965a6a68f | |
parent | 4b224e83655d6fad6d6711aaa2d5431866f14197 (diff) | |
parent | 93d17a11bd5016782ec3da59407917e8dcff879c (diff) |
Merge pull request #343 from NVIDIA/upstream
Relaxing CUDNN find param
-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() |