# IT4Innovations 2018

easyblock = 'PythonPackage'

name = 'Tensorflow'
version = '1.6.0rc0'
versionsuffix = '-CUDA-9.0.176'

homepage = 'https://www.tensorflow.org/'
description = "An open-source software library for Machine Intelligence"

toolchain = {'name': 'Py', 'version': '3.6'}

#source_urls = ['https://pypi.python.org/packages/1f/38/c255432f18005362dd2459e3bff2b2411eb7d02ca09fa20ee1dd1311f69f/']
sources = [
    'tensorflow-%(version)s-cp36-cp36m-linux_x86_64.whl',
]

dependencies = [
    ('cuDNN', '7.0.5', '-CUDA-9.0.176', True ),
    ('numpy', '1.13.3'),
    ('protobuf', '3.4.0'),
]

use_pip = True
unpack_sources = False

sanity_check_paths = {
    'files': ['bin/tensorboard'],
    'dirs': ['lib/python%(pyshortver)s/site-packages'],
}

moduleclass = 'lib'