torch>=1.9.0
torchvision>=0.10.0
numpy>=1.19.0
pynvml>=11.0.0
PyYAML>=5.4.1
tqdm>=4.62.0
pillow>=8.3.0
scikit-learn>=0.24.0
faiss-cpu>=1.7.0
datasets>=2.14.0
