numpy
torch>=1.12
