blob: de1324ff96e877c516cb0c6c6ceb6cf3bf8f6004 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
# Distributed under the OSI-approved BSD 3-Clause License. See accompanying
# file Copyright.txt or https://cmake.org/licensing for details.
#.rst:
# FindCUDNN
# -------
#
# Find CUDNN library
#
# Valiables that affect result:
# <VERSION>, <REQUIRED>, <QUIETLY>: as usual
#
# <EXACT> : as usual, plus we do find '5.1' version if you wanted '5'
# (not if you wanted '5.0', as usual)
#
# Result variables
# ^^^^^^^^^^^^^^^^
#
# This module will set the following variables in your project:
#
# ``CUDNN_INCLUDE``
# where to find cudnn.h.
# ``CUDNN_LIBRARY``
# the libraries to link against to use CUDNN.
# ``CUDNN_FOUND``
# If false, do not try to use CUDNN.
# ``CUDNN_VERSION``
# Version of the CUDNN library we looked for
#
# Exported functions
# ^^^^^^^^^^^^^^^^
# function(CUDNN_INSTALL version dest_dir)
# This function will try to download and install CUDNN.
#
#
function(CUDNN_INSTALL version dest_dir)
if(${CMAKE_SYSTEM_NAME} MATCHES "Linux")
if("${CMAKE_SYSTEM_PROCESSOR}" STREQUAL "x86_64")
set(__url_arch_name linux-x64 )
elseif("${CMAKE_SYSTEM_PROCESSOR}" MATCHES "ppc")
set(__url_arch_name linux-ppc64le )
# TX1 has to be installed via JetPack
endif()
elseif (APPLE)
set(__url_arch_name osx-x64)
elseif(WIN32)
if(CMAKE_SYSTEM_VERSION MATCHES "10")
set(__url_arch_name windows10)
else()
set(__url_arch_name windows7)
endif()
endif()
# Download and install CUDNN locally if not found on the system
if(__url_arch_name)
set(__download_dir ${CMAKE_CURRENT_BINARY_DIR}/downloads)
file(MAKE_DIRECTORY ${__download_dir})
set(__cudnn_filename cudnn-${CUDA_VERSION}-${__url_arch_name}-v${version}.tgz)
set(__base_url http://developer.download.nvidia.com/compute/redist/cudnn)
set(__cudnn_url ${__base_url}/v${version}/${__cudnn_filename})
set(__cudnn_tgz ${__download_dir}/${__cudnn_filename})
if(NOT EXISTS ${__cudnn_tgz})
message("Downloading CUDNN library from NVIDIA...")
file(DOWNLOAD ${__cudnn_url} ${__cudnn_tgz}
SHOW_PROGRESS STATUS CUDNN_STATUS
)
if("${CUDNN_STATUS}" MATCHES "0")
execute_process(COMMAND ${CMAKE_COMMAND} -E tar xzf "${__cudnn_tgz}" WORKING_DIRECTORY "${__download_dir}")
else()
message("Was not able to download CUDNN. Please install CuDNN manually from https://developer.nvidia.com/cuDNN")
endif()
endif()
if(WIN32)
file(GLOB __cudnn_binfiles ${__download_dir}/cuda/bin*/*)
install(FILES ${__cudnn_binfiles}
DESTINATION "${dest_dir}/bin")
endif()
file(GLOB __cudnn_incfiles ${__download_dir}/cuda/include/*)
install(FILES ${__cudnn_incfiles}
DESTINATION "${dest_dir}/include")
file(GLOB __cudnn_libfiles ${__download_dir}/cuda/lib*/*)
install(FILES ${__cudnn_libfiles}
DESTINATION "${dest_dir}/lib")
endif(__url_arch_name)
endfunction()
#####################################################
get_filename_component(__libpath_cudart ${CUDA_CUDART_LIBRARY} PATH)
unset(CUDNN_LIBRARY CACHE)
find_path(CUDNN_INCLUDE cudnn.h
PATHS ${CUDNN_PATH} $ENV{CUDNN_PATH} ${CUDA_TOOLKIT_INCLUDE} ENV{CMAKE_INCLUDE_PATH}
DOC "Path to CUDNN include directory." )
# We use major only in library search as major/minor is not entirely consistent among platforms.
# Also, looking for exact minor version of .so is in general not a good idea.
# More strict enforcement of minor/patch version is done if/when the header file is examined.
if(CUDNN_FIND_VERSION_EXACT)
SET(__cudnn_ver_suffix ".${CUDNN_FIND_VERSION_MAJOR}")
SET(__cudnn_lib_win_name cudnn64_${CUDNN_FIND_VERSION_MAJOR}.dll)
SET(CUDNN_MAJOR_VERSION ${CUDNN_FIND_MAJOR_VERSION})
else()
SET(__cudnn_lib_win_name cudnn64.dll)
endif()
find_library(CUDNN_LIBRARY NAMES libcudnn.so${__cudnn_ver_suffix} libcudnn${__cudnn_ver_suffix}.dylib ${__cudnn_lib_win_name}
PATHS $ENV{CUDNN_PATH} $ENV{LD_LIBRARY_PATH} ${__libpath_cudart}
DOC "CUDNN library." )
mark_as_advanced(CUDNN_INCLUDE CUDNN_LIBRARY )
if(CUDNN_INCLUDE)
file(READ ${CUDNN_INCLUDE}/cudnn.h CUDNN_VERSION_FILE_CONTENTS)
string(REGEX MATCH "define CUDNN_MAJOR * +([0-9]+)"
CUDNN_MAJOR_VERSION "${CUDNN_VERSION_FILE_CONTENTS}")
string(REGEX REPLACE "define CUDNN_MAJOR * +([0-9]+)" "\\1"
CUDNN_MAJOR_VERSION "${CUDNN_MAJOR_VERSION}")
string(REGEX MATCH "define CUDNN_MINOR * +([0-9]+)"
CUDNN_MINOR_VERSION "${CUDNN_VERSION_FILE_CONTENTS}")
string(REGEX REPLACE "define CUDNN_MINOR * +([0-9]+)" "\\1"
CUDNN_MINOR_VERSION "${CUDNN_MINOR_VERSION}")
string(REGEX MATCH "define CUDNN_PATCHLEVEL * +([0-9]+)"
CUDNN_PATCH_VERSION "${CUDNN_VERSION_FILE_CONTENTS}")
string(REGEX REPLACE "define CUDNN_PATCHLEVEL * +([0-9]+)" "\\1"
CUDNN_PATCH_VERSION "${CUDNN_PATCH_VERSION}")
endif()
if(NOT CUDNN_MAJOR_VERSION)
set(CUDNN_VERSION "???")
else()
## Fixing the case where 5.1 does not fit 'exact' 5.
set(CUDNN_VERSION ${CUDNN_MAJOR_VERSION}.${CUDNN_MINOR_VERSION})
if(CUDNN_FIND_VERSION_EXACT AND "x${CUDNN_FIND_VERSION_MINOR}" STREQUAL "x")
if(CUDNN_MAJOR_VERSION EQUAL CUDNN_FIND_VERSION_MAJOR)
set(CUDNN_VERSION ${CUDNN_FIND_VERSION})
endif()
endif()
math(EXPR CUDNN_VERSION_NUM "${CUDNN_MAJOR_VERSION} * 1000 + ${CUDNN_MINOR_VERSION} * 100 + ${CUDNN_PATCH_VERSION}")
message(STATUS "Found Cudnn_${CUDNN_VERSION_NUM} at ${CUDNN_INCLUDE} ${CUDNN_LIBRARY}")
endif()
find_package_handle_standard_args(CUDNN
REQUIRED_VARS CUDNN_LIBRARY
VERSION_VAR CUDNN_VERSION)
|
