numpy
torch>=1.4.0
