diff --git a/easybuild/easyconfigs/b/bitsandbytes/bitsandbytes-0.46.1-foss-2024a.eb b/easybuild/easyconfigs/b/bitsandbytes/bitsandbytes-0.46.1-foss-2024a.eb new file mode 100644 index 000000000000..539fe0abdda8 --- /dev/null +++ b/easybuild/easyconfigs/b/bitsandbytes/bitsandbytes-0.46.1-foss-2024a.eb @@ -0,0 +1,52 @@ +easyblock = 'CMakeMake' + +name = 'bitsandbytes' +version = '0.46.1' + +homepage = 'https://huggingface.co/docs/bitsandbytes/main/en/index' +description = "bitsandbytes enables accessible large language models via k-bit quantization for PyTorch." +github_account = 'bitsandbytes-foundation' + +toolchain = {'name': 'foss', 'version': '2024a'} + +source_urls = [GITHUB_LOWER_SOURCE] +sources = ['%(version)s.tar.gz'] +checksums = ['8326835082ad5590e4eab2cc51129bf55dd1c16e3d3038bc23431371c24b47da'] + +builddependencies = [ + ('CMake', '3.29.3'), + ('pkgconf', '2.2.0'), + ('scikit-build-core', '0.10.6'), +] + +dependencies = [ + ('Python', '3.12.3'), + ('PyTorch', '2.6.0'), + ('SciPy-bundle', '2024.05'), +] + +configopts = "-DCOMPUTE_BACKEND=cpu" + +# skip install step in CMakeMake +skipsteps = ['install'] + +exts_defaultclass = 'PythonPackage' +exts_list = [ + (name, version, { + 'source_urls': ['https://github.com/%(github_account)s/%(namelower)s/archive'], + 'sources': ['%(version)s.tar.gz'], + 'start_dir': '%(builddir)s/bitsandbytes-%(version)s/', + 'checksums': ['8326835082ad5590e4eab2cc51129bf55dd1c16e3d3038bc23431371c24b47da'], + }), +] + +sanity_check_paths = { + 'files': [f'lib/python%(pyshortver)s/site-packages/bitsandbytes/libbitsandbytes_cpu.{SHLIB_EXT}'], + 'dirs': ['lib/python%(pyshortver)s/site-packages'], +} + +sanity_check_commands = [ + "python -s -c 'import bitsandbytes'", +] + +moduleclass = 'ai' diff --git a/easybuild/easyconfigs/p/PyTorch-Lightning/PyTorch-Lightning-2.5.2-foss-2024a.eb b/easybuild/easyconfigs/p/PyTorch-Lightning/PyTorch-Lightning-2.5.2-foss-2024a.eb new file mode 100644 index 000000000000..6c5ef354db6e --- /dev/null +++ b/easybuild/easyconfigs/p/PyTorch-Lightning/PyTorch-Lightning-2.5.2-foss-2024a.eb @@ -0,0 +1,40 @@ +easyblock = 'PythonBundle' + +name = 'PyTorch-Lightning' +version = '2.5.2' + +homepage = 'https://pytorchlightning.ai' +description = "PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers." + +toolchain = {'name': 'foss', 'version': '2024a'} + +dependencies = [ + ('Python', '3.12.3'), + # dependencies from requirements/pytorch/base.txt + ('PyTorch', '2.6.0'), + ('tqdm', '4.66.5'), + ('aiohttp', '3.10.10'), + ('typing-extensions', '4.11.0'), + # optional dependencies from requirements/pytorch/extra.txt + ('bitsandbytes', '0.46.1'), + ('tensorboardX', '2.6.4'), + ('matplotlib', '3.9.2'), + ('Hydra', '1.3.2'), +] + +# remaining dependencies from requirements/pytorch/base.txt +exts_list = [ + ('lightning-utilities', '0.14.3', { + 'source_tmpl': 'lightning_utilities-%(version)s.tar.gz', + 'checksums': ['37e2f83f273890052955a44054382c211a303012ee577619efbaa5df9e65e9f5'], + }), + ('torchmetrics', '1.7.3', { + 'checksums': ['08450a19cdb67ba1608aac0b213e5dc73033e11b60ad4719696ebcede591621e'], + }), + ('pytorch-lightning', version, { + 'source_tmpl': 'pytorch_lightning-%(version)s.tar.gz', + 'checksums': ['f817087d611be8d43b777dd4e543d72703e235510936677a13e6c29f7fd790e3'], + }), +] + +moduleclass = 'ai'