mirror of
https://code.it4i.cz/sccs/easyconfigs-it4i.git
synced 2025-04-14 02:41:22 +01:00
5312a529a4
modified: c/CP2K/CP2K-5.1.eb modified: h/horovod/horovod-0.11.3-Py-3.6.eb new file: m/MotionCor2/MotionCor2-1.0.5-CUDA-8.0.44.eb new file: m/mpi4py/mpi4py-3.0.0-Py-3.6.eb
34 lines
772 B
Plaintext
34 lines
772 B
Plaintext
# 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 = [
|
|
('Keras', '2.1.4', '-Tensorflow-1.6.0rc0-CUDA-9.0.176'),
|
|
('OpenMPI', '1.10.7', '', ('GCC', '6.3.0-2.27')),
|
|
('NCCL', '2.1.4', '', True)
|
|
]
|
|
|
|
prebuildopts = 'HOROVOD_NCCL_HOME=/apps/all/NCCL/2.1.4 HOROVOD_GPU_ALLREDUCE=NCCL '
|
|
|
|
modextravars = {
|
|
'HOROVOD_NCCL_HOME': '/apps/all/NCCL/2.1.4',
|
|
'HOROVOD_GPU_ALLREDUCE': 'NCCL',
|
|
}
|
|
|
|
sanity_check_paths = {
|
|
'files': [],
|
|
'dirs': ['lib/python3.6/site-packages'],
|
|
}
|
|
|
|
moduleclass = 'python'
|