set cudnn version to 7 and REQUIRED (#755)

NervanaSystems · Mar 26, 2018 · 8364633 · 8364633
1 parent a29f675
commit 8364633
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -147,7 +147,7 @@ find_package(CUDA 8 QUIET)
 if(CUDA_FOUND AND NGRAPH_GPU_ENABLE)
     message(STATUS "GPU Backend Enabled")
     set(NGRAPH_GPU_ENABLE TRUE)
-    find_package(CUDNN 5 QUIET REQUIRED)
+    find_package(CUDNN 7 REQUIRED)
     include_directories(SYSTEM ${CUDA_INCLUDE_DIRS} ${CUDNN_INCLUDE_DIR} ${LLVM_INCLUDE_DIR})
     if (CMAKE_C_COMPILER_ID STREQUAL "GNU" AND
         NOT CMAKE_C_COMPILER_VERSION VERSION_LESS 6.0 AND