horovod-0.11.3-Py-3.6.eb 772 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
# IT4Innovations 2018

easyblock = "PythonPackage"
name = 'horovod'
version = '0.11.3'

homepage = 'https://pypi.python.org/pypi/horovod'
description = 'Distributed training framework for TensorFlow.'

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

source_urls = [PYPI_SOURCE]
sources = [SOURCE_TAR_GZ]

dependencies = [
16 17
    ('Keras', '2.1.4', '-Tensorflow-1.6.0rc0-CUDA-9.0.176'),
    ('OpenMPI', '1.10.7', '', ('GCC', '6.3.0-2.27')),
18
    ('NCCL', '2.1.4', '', True)
19 20
]

21 22
prebuildopts = 'HOROVOD_NCCL_HOME=/apps/all/NCCL/2.1.4 HOROVOD_GPU_ALLREDUCE=NCCL '

23
modextravars = {
24
    'HOROVOD_NCCL_HOME': '/apps/all/NCCL/2.1.4',
25 26 27
    'HOROVOD_GPU_ALLREDUCE': 'NCCL',
}

28 29
sanity_check_paths = {
    'files': [],
30
    'dirs': ['lib/python3.6/site-packages'],
31 32 33
}

moduleclass = 'python'