File tree Expand file tree Collapse file tree 1 file changed +11
-7
lines changed Expand file tree Collapse file tree 1 file changed +11
-7
lines changed Original file line number Diff line number Diff line change @@ -10,20 +10,24 @@ else()
10
10
CACHE PATH "CUDNN ROOT" )
11
11
endif ()
12
12
13
+ set (TARGET_ARCH "x86_64" )
14
+ if (NOT ${CMAKE_SYSTEM_PROCESSOR} )
15
+ set (TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR} )
16
+ endif ()
17
+
13
18
find_path (
14
19
CUDNN_INCLUDE_DIR cudnn.h
15
- PATHS ${CUDNN_ROOT} ${CUDNN_ROOT} /include $ENV{CUDNN_ROOT}
16
- $ENV{CUDNN_ROOT} /include ${CUDA_TOOLKIT_INCLUDE}
20
+ PATHS ${CUDNN_ROOT}
21
+ ${CUDNN_ROOT} /include
22
+ ${CUDNN_ROOT} /include/${TARGET_ARCH}-linux-gnu
23
+ $ENV{CUDNN_ROOT}
24
+ $ENV{CUDNN_ROOT} /include
25
+ ${CUDA_TOOLKIT_INCLUDE}
17
26
/usr/local/lib/python${PY_VERSION}/dist-packages/nvidia/cudnn/include/
18
27
NO_DEFAULT_PATH )
19
28
20
29
get_filename_component (__libpath_hist ${CUDA_CUDART_LIBRARY} PATH )
21
30
22
- set (TARGET_ARCH "x86_64" )
23
- if (NOT ${CMAKE_SYSTEM_PROCESSOR} )
24
- set (TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR} )
25
- endif ()
26
-
27
31
list (
28
32
APPEND
29
33
CUDNN_CHECK_LIBRARY_DIRS
You can’t perform that action at this time.
0 commit comments