numpy>=1.20.0
torch>=1.8.0
