numpy~=1.26.0 h5py~=3.9.0 faiss-gpu==1.7.2 torch