Skip to content

N
ngraph
Commit 8364633b authored by Fenglei's avatar Fenglei Committed by Scott Cyphers
Browse files
Options

set cudnn version to 7 and REQUIRED (#755)

parent a29f6754
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
CMakeLists.txt
View file @ 8364633b
...@@ -147,7 +147,7 @@ find_package(CUDA 8 QUIET) ...@@ -147,7 +147,7 @@ find_package(CUDA 8 QUIET)
if(CUDA_FOUND AND NGRAPH_GPU_ENABLE) if(CUDA_FOUND AND NGRAPH_GPU_ENABLE)
message(STATUS "GPU Backend Enabled") message(STATUS "GPU Backend Enabled")
set(NGRAPH_GPU_ENABLE TRUE) 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}) include_directories(SYSTEM ${CUDA_INCLUDE_DIRS} ${CUDNN_INCLUDE_DIR} ${LLVM_INCLUDE_DIR})
if (CMAKE_C_COMPILER_ID STREQUAL "GNU" AND if (CMAKE_C_COMPILER_ID STREQUAL "GNU" AND
NOT CMAKE_C_COMPILER_VERSION VERSION_LESS 6.0 AND NOT CMAKE_C_COMPILER_VERSION VERSION_LESS 6.0 AND
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment