torch >= 1.9.0
torchvision
numpy
tqdm
tensorboardX
pickle