CuPy: NumPy & SciPy for GPU
conda install main::cupy
CuPy is an open-source array library for GPU-accelerated computing with Python. It provides GPU versions of many NumPy and SciPy functions.