modified: c/CUDA/CUDA-9.1.85.eb

new file:   c/cuDNN/cuDNN-7.1-CUDA-9.1.85.eb
This commit is contained in:
Lukáš Krupčík 2018-03-12 11:30:41 +01:00
parent 78c5694293
commit 9c890aa58e
2 changed files with 27 additions and 1 deletions

View File

@ -11,7 +11,7 @@ toolchain = {'name': 'dummy', 'version': ''}
source_urls = ['https://developer.nvidia.com/compute/cuda/%(version_major_minor)s/prod/local_installers/']
sources = ['%(namelower)s_%(version)s_linux-run']
sources = ['cuda_9.1.85_387.26_linux']
modextravars = { 'CUDA_HOME': '%(installdir)s' }

View File

@ -0,0 +1,26 @@
easyblock = 'Tarball'
name = 'cuDNN'
version = '7.41'
cuda_version = '9.1.85'
versionsuffix = '-CUDA-%s' % cuda_version
homepage = 'https://developer.nvidia.com/cudnn'
description = """The NVIDIA CUDA Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for
deep neural networks."""
toolchain = {'name': 'dummy', 'version': ''}
# Nvidia developer registration required.
# Download link: https://developer.nvidia.com/rdp/cudnn-download
sources = ['cudnn-9.1-linux-x64-v7.1.tgz']
dependencies = [('CUDA', cuda_version)]
sanity_check_paths = {
'files': ['include/cudnn.h', 'lib64/libcudnn_static.a'],
'dirs': ['include', 'lib64'],
}
moduleclass = 'numlib'