numpy
cupy-cuda11x

