diff --git a/easybuild/easyconfigs/c/captum/captum-0.8.0-foss-2024a-CUDA-12.6.0.eb b/easybuild/easyconfigs/c/captum/captum-0.8.0-foss-2024a-CUDA-12.6.0.eb new file mode 100644 index 00000000000..fb7cd031432 --- /dev/null +++ b/easybuild/easyconfigs/c/captum/captum-0.8.0-foss-2024a-CUDA-12.6.0.eb @@ -0,0 +1,33 @@ +easyblock = 'PythonBundle' + +name = 'captum' +version = '0.8.0' +versionsuffix = '-CUDA-%(cudaver)s' + +homepage = 'https://captum.ai/' +description = """ +Captum is a model interpretability and understanding library for PyTorch. +Captum means comprehension in Latin and contains general purpose +implementations of integrated gradients, saliency maps, smoothgrad, vargrad and +others for PyTorch models. It has quick integration for models built with +domain-specific libraries such as torchvision, torchtext, and others. +""" + +toolchain = {'name': 'foss', 'version': '2024a'} + +dependencies = [ + ('CUDA', '12.6.0', '', SYSTEM), + ('Python', '3.12.3'), + ('PyTorch', '2.7.1', '-CUDA-%(cudaver)s'), + ('matplotlib', '3.9.2'), + ('tqdm', '4.66.5'), + ('scikit-learn', '1.5.2'), +] + +exts_list = [ + (name, version, { + 'checksums': ['73f2e5a07ffe6ac6647a6f12cd2f67f0d8f717329f150ddf08ef73cba0858d26'], + }), +] + +moduleclass = 'ai'