CuPy is a Python library compatible with NumPy/SciPy, designed for GPU-accelerated computing. It supports NVIDIA CUDA and AMD ROCm platforms, offering functionality such as low-level CUDA features and direct CUDA Runtime API calls. Installation packages are available via PyPI and Conda-Forge for various architectures. CuPy also allows containerized execution with NVIDIA Container Toolkit.
2 Comments
Sort: