# NOTE: Previous versions of pytorch and torchvision might also work as well,
# but we haven't test them yet
torch>=1.7.1
torchvision>=0.8.2
matplotlib
numpy
tensorboard