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。