CuPy is an implementation of a NumPy-compatible multi-dimensional array on CUDA.
conda install anaconda::cupy