torch>=2.2
torchvision>=0.17
tqdm
numpy