torch>=1.10.0
matplotlib
tqdm
torchsummary
