torch>=1.1.0
numpy<=1.18.5
cupy-cuda102
