torch>=2.0.0
numpy
nvidia-cuda-runtime-cu12
