torch>=2.5.0
numpy
tqdm
pytest
pillow
pandas
torchvision
pyarrow