numpy>=1.16.0

[cuda]
torch>=1.7.0
