torch>=1.3
numpy
scipy
cupy>=7.0.0
