CuPy is an implementation of NumPy-compatible multi-dimensional array on CUDA
conda install npbool::cupy
CuPy consists of cupy.ndarray, the core multi-dimensional array class, and many functions on it. It supports a subset of numpy.ndarray interface.