hf/transformers/examples/pytorch/image-classification/requirements.txt

accelerate>=0.12.0
torch>=1.5.0
torchvision>=0.6.0
datasets>=1.17.0
evaluate