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