torch>=1.1
numpy
