Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
easyblock = 'PythonBundle'

name = 'Horovod'
version = '0.28.1'
local_pt_version = '1.11.0'
local_cuda_suffix = '-CUDA-%(cudaver)s'
versionsuffix = local_cuda_suffix + '-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': '2021a'}

builddependencies = [
('CMake', '3.20.1'),
]
dependencies = [
('Python', '3.9.5'),
('PyYAML', '5.4.1'),
('CUDA', '11.3.1', '', SYSTEM),
('NCCL', '2.10.3', local_cuda_suffix),
('PyTorch', local_pt_version, local_cuda_suffix),
]

use_pip = True
sanity_pip_check = True

preinstallopts = 'HOROVOD_WITH_MPI=1 HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL '
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'
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
easyblock = 'PythonBundle'

name = 'Horovod'
version = '0.28.1'
local_pt_version = '1.12.1'
local_cuda_suffix = '-CUDA-%(cudaver)s'
versionsuffix = local_cuda_suffix + '-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': '2021a'}

builddependencies = [
('CMake', '3.20.1'),
]
dependencies = [
('Python', '3.9.5'),
('PyYAML', '5.4.1'),
('CUDA', '11.3.1', '', SYSTEM),
('NCCL', '2.10.3', local_cuda_suffix),
('PyTorch', local_pt_version, local_cuda_suffix),
]

use_pip = True
sanity_pip_check = True

preinstallopts = 'HOROVOD_WITH_MPI=1 HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL '
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'
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
easyblock = 'PythonBundle'

name = 'Horovod'
version = '0.28.1'
local_pt_version = '1.12.1'
local_cuda_suffix = '-CUDA-%(cudaver)s'
versionsuffix = local_cuda_suffix + '-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': '2021b'}

builddependencies = [
('CMake', '3.21.1'),
]
dependencies = [
('Python', '3.9.6'),
('PyYAML', '5.4.1'),
('CUDA', '11.5.2', '', SYSTEM),
('NCCL', '2.10.3', local_cuda_suffix),
('PyTorch', local_pt_version, local_cuda_suffix),
]

use_pip = True
sanity_pip_check = True

preinstallopts = 'HOROVOD_WITH_MPI=1 HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL '
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'
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
easyblock = 'PythonBundle'

name = 'Horovod'
version = '0.28.1'
local_pt_version = '1.12.0'
local_cuda_suffix = '-CUDA-%(cudaver)s'
versionsuffix = local_cuda_suffix + '-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'),
('CUDA', '11.7.0', '', SYSTEM),
('NCCL', '2.12.12', local_cuda_suffix),
('PyTorch', local_pt_version, local_cuda_suffix),
]

use_pip = True
sanity_pip_check = True

preinstallopts = 'HOROVOD_WITH_MPI=1 HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL '
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'
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
easyblock = 'PythonBundle'

name = 'Horovod'
version = '0.28.1'
local_pt_version = '1.12.1'
local_cuda_suffix = '-CUDA-%(cudaver)s'
versionsuffix = local_cuda_suffix + '-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'),
('CUDA', '11.7.0', '', SYSTEM),
('NCCL', '2.12.12', local_cuda_suffix),
('PyTorch', local_pt_version, local_cuda_suffix),
]

use_pip = True
sanity_pip_check = True

preinstallopts = 'HOROVOD_WITH_MPI=1 HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL '
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'
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
easyblock = 'PythonBundle'

name = 'Horovod'
version = '0.28.1'
local_pt_version = '1.9.0'
versionsuffix = '-PyTorch-%s' % local_pt_version

homepage = 'https://github.com/uber/horovod'
description = "Horovod is a distributed training framework for TensorFlow."

toolchain = {'name': 'fosscuda', 'version': '2020b'}

builddependencies = [
('CMake', '3.18.4'),
]
dependencies = [
('Python', '3.8.6'),
('PyYAML', '5.3.1'),
('NCCL', '2.8.3', '-CUDA-%(cudaver)s'),
('PyTorch', local_pt_version),
]

use_pip = True
sanity_pip_check = True

preinstallopts = 'HOROVOD_WITH_MPI=1 HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL '
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'],
}

moduleclass = 'tools'