torch>=0.4.1
numpy
