numpy
nvidia-cuda-runtime
nvidia-curand
nvidia-cublas
