numpy
torch>=1.0
tensorboard
