torch>=2.3.1
torchdata==0.9.0
tqdm
