FAISS#

FAISS v1.8 提供了一个特殊的 conda 包,可以为 GPU 上的 Flat、IVF-Flat 和 IVF-PQ 索引启用 RAFT 后端。与经典的 FAISS GPU 索引一样,RAFT 后端还支持 FAISS CPU 索引之间的互操作性,允许索引在 GPU 上训练,在 CPU 上搜索,反之亦然。

通过从源代码构建 FAISS 并启用 FAISS_USE_RAFT cmake 标志,以及为启用 RAFT 的 GPU 索引设置 use_raft 配置选项,可以启用 RAFT 后端。

也可以使用以下命令安装预编译的 conda 包

conda install -c conda-forge -c pytorch -c rapidsai -c nvidia -c "nvidia/label/cuda-11.8.0" faiss-gpu-raft

随着我们将矢量搜索算法从 RAFT 迁移到 cuVS,FAISS 的下一个版本将支持 cuVS。