torch>=2.3.1
torchdata>=0.11.0
tqdm
