new file: c/cuDNN/cuDNN-7.0.5-CUDA-9.0.176.eb

new file:   t/Tensorflow/Tensorflow-1.6.0rc0-CUDA-9.0.176-Python-3.6.1.eb
This commit is contained in:
Lukáš Krupčík 2018-02-09 09:09:47 +01:00
parent 9e2aecf93a
commit 0db84c282b
2 changed files with 65 additions and 0 deletions

View File

@ -0,0 +1,30 @@
easyblock = 'Tarball'
name = 'cuDNN'
version = '7.0.5'
cuda_version = '9.0.176'
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.0-linux-x64-v7.tar.gz']
dependencies = [('CUDA', cuda_version)]
#checksums = [
# '099ca6a9b21b35e32f3408eff2a1c06d', # cudnn-8.0-linux-x64-v5.1.tgz
#]
sanity_check_paths = {
'files': ['include/cudnn.h', 'lib64/libcudnn_static.a'],
'dirs': ['include', 'lib64'],
}
moduleclass = 'numlib'

View File

@ -0,0 +1,35 @@
easyblock = 'PythonPackage'
name = 'Tensorflow'
version = '1.6.0rc0'
versionsuffix = '-CUDA-9.0.176-Python-%(pyver)s'
homepage = 'https://www.tensorflow.org/'
description = "An open-source software library for Machine Intelligence"
toolchain = {'name': 'dummy', 'version': ''}
pymaj = '3'
pymin = '6'
pymajmin = pymaj + pymin
#source_urls = ['https://pypi.python.org/packages/1f/38/c255432f18005362dd2459e3bff2b2411eb7d02ca09fa20ee1dd1311f69f/']
sources = [
'tensorflow-%(version)s-cp36-cp36m-linux_x86_64.whl',
]
dependencies = [
('Python', '%s.%s.1' % (pymaj, pymin)),
('cuDNN', '7.0.5', '-CUDA-9.0.176' ),
('GCC', '6.3.0-2.27'),
]
use_pip = True
unpack_sources = False
sanity_check_paths = {
'files': ['bin/tensorboard'],
'dirs': ['lib/python%(pyshortver)s/site-packages'],
}
moduleclass = 'lib'