faiss-cpu >= 1.7.0
datasets >= 1.6.2
psutil >= 5.7.0
torch >= 1.4.0
pytorch-lightning ==  1.3.1
nvidia-ml-py3 == 7.352.0
ray >=  1.3.0