# IT4Innovation 2017

easyblock = 'PythonPackage'

name = 'CNTK'
version = '2.0'
versionsuffix = '-Python-%(local_pyver)s'

homepage = 'https://cntk.ai/'
description = """The Microsoft Cognitive Toolkit is a unified deep-learning toolkit that describes neural networks as a series of local_computational steps via a directed graph. In this directed graph, leaf nodes represent input values or network parameters, while other nodes represent matrix operations upon their inputs. CNTK allows to easily realize and combine popular model types such as feed-forward DNNs, convolutional nets (CNNs), and recurrent networks (RNNs/LSTMs). It implements stochastic gradient descent (SGD, error backpropagation) learning with automatic differentiation and parallelization across multiple GPUs and servers. """

toolchain = {'name': 'GCC', 'version': '6.3.0-2.27'}

source_urls = ['https://cntk.ai/PythonWheel/CPU-Only']
sources = ['cntk-2.0-cp36-cp36m-linux_x86_64.whl']

dependencies = [
    ('Python', '3.6.1', '', True),
    ('OpenMPI', '2.1.0')
]

unpack_sources = False

use_pip = True

skipsteps = ['sanitycheck']

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

moduleclass = 'math'