Skip to content

Commit

Permalink
adding easyconfigs: Horovod-0.28.1-foss-2022a-PyTorch-1.12.0.eb
Browse files Browse the repository at this point in the history
  • Loading branch information
casparl committed Nov 7, 2023
1 parent c757d75 commit 121dd59
Showing 1 changed file with 45 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -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'

0 comments on commit 121dd59

Please sign in to comment.