matplotlib
numpy
pandas
Pillow
scipy
six
tqdm
tensorboard==2.2.0
torch
torchvision