numpy
torch
torchvision
matplotlib>=3.0.0
Pillow
scipy
tqdm
psutil
tensorboardX
