matplotlib
numpy
pandas
Pillow
scipy
six
tqdm
tensorboard>=2.3.0
torch
torchvision