torchvision==0.14.1
scikit-learn
matplotlib
tqdm
Pillow
examples-utils[common] @ git+https://github.com/graphcore/examples-utils@latest_stable