diff --git a/easybuild/easyconfigs/z/Zennit/Zennit-1.0.0-foss-2024a-CUDA-12.6.0.eb b/easybuild/easyconfigs/z/Zennit/Zennit-1.0.0-foss-2024a-CUDA-12.6.0.eb new file mode 100644 index 00000000000..e64f8dffb33 --- /dev/null +++ b/easybuild/easyconfigs/z/Zennit/Zennit-1.0.0-foss-2024a-CUDA-12.6.0.eb @@ -0,0 +1,36 @@ +easyblock = 'PythonBundle' + +name = 'Zennit' +version = '1.0.0' +versionsuffix = '-CUDA-%(cudaver)s' + +homepage = 'https://zennit.readthedocs.io' +description = """Zennit (Zennit explains neural networks in torch) is a high-level framework in +Python using Pytorch for explaining/exploring neural networks. Its design +philosophy is intended to provide high customizability and integration as a +standardized solution for applying rule-based attribution methods in research, +with a strong focus on Layerwise Relevance Propagation (LRP).""" + +toolchain = {'name': 'foss', 'version': '2024a'} + +builddependencies = [('hatchling', '1.24.2')] + +dependencies = [ + ('CUDA', '12.6.0', '', SYSTEM), + ('Python', '3.12.3'), + ('PyTorch', '2.6.0', versionsuffix), + ('torchvision', '0.21.0', versionsuffix), + ('Pillow', '10.4.0'), +] + +use_pip = True + +exts_list = [ + ('zennit', version, { + 'checksums': ['c6772100035172313d30986e5e65bc09327dc614fa8d2e9a155a465d3c83355d'], + }), +] + +sanity_pip_check = True + +moduleclass = 'vis' diff --git a/easybuild/easyconfigs/z/Zennit/Zennit-1.0.0-foss-2024a.eb b/easybuild/easyconfigs/z/Zennit/Zennit-1.0.0-foss-2024a.eb new file mode 100644 index 00000000000..2c97e14dbc8 --- /dev/null +++ b/easybuild/easyconfigs/z/Zennit/Zennit-1.0.0-foss-2024a.eb @@ -0,0 +1,34 @@ +easyblock = 'PythonBundle' + +name = 'Zennit' +version = '1.0.0' + +homepage = 'https://zennit.readthedocs.io' +description = """Zennit (Zennit explains neural networks in torch) is a high-level framework in +Python using Pytorch for explaining/exploring neural networks. Its design +philosophy is intended to provide high customizability and integration as a +standardized solution for applying rule-based attribution methods in research, +with a strong focus on Layerwise Relevance Propagation (LRP).""" + +toolchain = {'name': 'foss', 'version': '2024a'} + +builddependencies = [('hatchling', '1.24.2')] + +dependencies = [ + ('Python', '3.12.3'), + ('PyTorch', '2.6.0'), + ('torchvision', '0.21.0'), + ('Pillow', '10.4.0'), +] + +use_pip = True + +exts_list = [ + ('zennit', version, { + 'checksums': ['c6772100035172313d30986e5e65bc09327dc614fa8d2e9a155a465d3c83355d'], + }), +] + +sanity_pip_check = True + +moduleclass = 'vis'