hf/transformers/examples/research_projects/rag-end2end-retriever/requirements.txt

faiss-cpu >= 1.7.2
datasets 
psutil >= 5.9.1
torch >= 1.11.0
pytorch-lightning == 1.6.4
nvidia-ml-py3 == 7.352.0
ray >=  1.13.0