numpy
torch>=1.0.0
