numpy
torch >= 1.0.0
