MONAI-1.0.1-foss-2022a.eb

easyblock = 'PythonBundle'

name = 'MONAI'
version = '1.0.1'

homepage = 'https://monai.io/'
description = """
MONAI is a PyTorch-based, open-source framework for deep learning in healthcare
imaging, part of PyTorch Ecosystem.
"""

toolchain = {'name': 'foss', 'version': '2022a'}

github_account = 'Project-MONAI'

dependencies = [
    ('Python', '3.10.4'),
    ('PyTorch', '1.12.0'),
    ('einops', '0.4.1'),
    ('ITK', '5.2.1'),
    ('NiBabel', '4.0.2'),
    ('scikit-image', '0.19.3'),
    ('tensorboard', '2.10.0'),
    ('torchvision', '0.13.1'),
    ('tqdm', '4.64.0'),
]

exts_list = [
    (name, version, {
        'preinstallopts': 'BUILD_MONAI=1',
        'source_tmpl': '%(version)s.tar.gz',
        'source_urls': ['https://github.com/%(github_account)s/%(name)s/archive'],
        'checksums': [('dc961e0adbe1b0c87b85f69377267d60c835f24eb03c8982a21b23a93c6ec36b',
                       'bb1a84ea3c4cd2c3756f72359a891843e0efa02da17537fb158b28c4e0b2d243')],
    }),
]

use_pip = True

sanity_pip_check = True

sanity_check_commands = ["python -c 'import monai; monai.config.print_config()'"]

sanity_check_paths = {
    'files': ['lib/python%%(pyshortver)s/site-packages/%%(namelower)s/_C.%s' % SHLIB_EXT],
    'dirs': [],
}

moduleclass = 'vis'