einops
faiss-gpu
numpy
torch==1.6
tqdm
