From 121dd592afa1f7fe9d65cf9a0052c00222172cb1 Mon Sep 17 00:00:00 2001 From: casparl Date: Tue, 7 Nov 2023 15:01:16 +0100 Subject: [PATCH] adding easyconfigs: Horovod-0.28.1-foss-2022a-PyTorch-1.12.0.eb --- ...orovod-0.28.1-foss-2022a-PyTorch-1.12.0.eb | 45 +++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 easybuild/easyconfigs/h/Horovod/Horovod-0.28.1-foss-2022a-PyTorch-1.12.0.eb diff --git a/easybuild/easyconfigs/h/Horovod/Horovod-0.28.1-foss-2022a-PyTorch-1.12.0.eb b/easybuild/easyconfigs/h/Horovod/Horovod-0.28.1-foss-2022a-PyTorch-1.12.0.eb new file mode 100644 index 00000000000..089c87eaf0a --- /dev/null +++ b/easybuild/easyconfigs/h/Horovod/Horovod-0.28.1-foss-2022a-PyTorch-1.12.0.eb @@ -0,0 +1,45 @@ +easyblock = 'PythonBundle' + +name = 'Horovod' +version = '0.28.1' +local_pt_version = '1.12.0' +versionsuffix = '-PyTorch-%s' % local_pt_version + +homepage = 'https://github.com/uber/horovod' +description = """Horovod is a distributed training framework for TensorFlow, PyTorch and MXnet. +This build only has PyTorch enabled.""" + +toolchain = {'name': 'foss', 'version': '2022a'} + +builddependencies = [ + ('CMake', '3.23.1'), +] +dependencies = [ + ('Python', '3.10.4'), + ('PyYAML', '6.0'), + ('PyTorch', local_pt_version), +] + +use_pip = True +sanity_pip_check = True + +preinstallopts = 'HOROVOD_WITH_MPI=1 ' +preinstallopts += 'HOROVOD_WITHOUT_TENSORFLOW=1 HOROVOD_WITH_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1 ' + +exts_list = [ + ('cloudpickle', '2.2.1', { + 'checksums': ['d89684b8de9e34a2a43b3460fbca07d09d6e25ce858df4d5a44240403b6178f5'], + }), + ('horovod', version, { + 'checksums': ['92a43f5a94c43907a56805bad15f19700c62ffc83b7ca483f9e104e229f67ef0'], + }), +] + +sanity_check_paths = { + 'files': ['bin/horovodrun'], + 'dirs': ['lib/python%(pyshortver)s/site-packages'], +} + +sanity_check_commands = ["horovodrun --help"] + +moduleclass = 'tools'