This is where the good new CuPy library appeared in CuPy . CuPy — it is a NumPy compatible library for GPUs.