torch>=1.9.0
matplotlib
tqdm
torchsummary
