Skip to content

[CuPy] [package list]

CuPy/11.0.0-cpeGNU-22.08 (CuPy-11.0.0-cpeGNU-22.08.eb)

Install with the EasyBuild-user module: 

eb CuPy-11.0.0-cpeGNU-22.08.eb -r
To access module help after installation and get reminded for which stacks and partitions the module is installed, use module spider CuPy/11.0.0-cpeGNU-22.08.

EasyConfig:

easyblock = 'PythonPackage'

name = 'CuPy'
version = '11.0.0'

homepage = 'https://cupy.dev'

whatis = [
    'Description: CuPy is an open-source array library for GPU-accelerated computing with Python'
]

description = """"
CuPy is an open-source array library for GPU-accelerated computing with Python.
CuPy acts as a drop-in replacement to run existing NumPy/SciPy code on NVIDIA CUDA 
or AMD ROCm platforms.
"""

toolchain = {'name': 'cpeGNU', 'version': '22.08'}

source_urls = [PYPI_LOWER_SOURCE]
sources = [SOURCELOWER_TAR_GZ]

dependencies = [
    ('cray-python', EXTERNAL_MODULE),
    ('rocm/5.0.2', EXTERNAL_MODULE)
]

use_pip = True

preinstallopts = ('export ROCM_HOME=$CRAY_ROCM_DIR && '
                  'export CUPY_INSTALL_USE_HIP=1 && '
                  'export HCC_AMDGPU_TARGET=gfx90a && '
                  )

skipsteps = ['build']

exts_defaultclass = 'PythonPackage'
exts_default_options = {
    'source_urls': [PYPI_SOURCE],
    'use_pip': True
}
exts_list = [
    ('fastrlock', '0.8')
]

sanity_check_paths = {
    'files': ['lib/python%(pyshortver)s/site-packages/%(namelower)s/cusolver.cpython-39-x86_64-linux-gnu.so',
              'lib/python%(pyshortver)s/site-packages/fastrlock/rlock.cpython-39-x86_64-linux-gnu.so'],
    'dirs': ['lib/python%(pyshortver)s/site-packages'],
}

moduleclass = 'devel'

[CuPy] [package list]