torch>=1.10
numpy
tensorboard
