cupy: a Python array library accelerated with NVIDIA CUDA
Overview of package
| Package: | cupy |
|---|---|
| Description: | a Python array library accelerated with NVIDIA CUDA |
| For more information: | https://cupy.dev/ |
| Categories: | PythonModule |
| License: | Unknown |
General Usage
CuPy is an open-source array library accelerated with NVIDIA CUDA. CuPy provides GPU accelerated computing with Python, using CUDA-related libraries including cuBLAS, cuDNN, cuRand, cuSolver, cuSPARSE, cuFFT, cuTensor and NCCL.
This module adds cuPy to PYTHONPATH and defines the environment variables CUTENSOR_LIBDIR and CUTENSOR_INCDIR
Available versions by cluster
This section lists the available versions of the package cupyon the UMD HPC clusters.
Versions available on the Zaratan cluster
| Version | Module tags | CPU(s) optimized for | GPU ready? |
|---|---|---|---|
| 11.5.0 | 11.5.0 | zen2 | Yes |