Bootstrap

Windows利用conda安装gpu版本Faiss + Ubuntu源码安装Faiss-gpu 记录(待更新~)

前言

由于在cpu上使用对向量检索算法时,发现面对数据量较大时,批量匹配耗时会显著增加,影响业务整体响应。便尝试使用GPU来实现检索计算,限于本人技术有限,写不出好算法。便取巧利用Faiss-gpu来检索(* ^ ▽ ^ *)
在这里插入图片描述
以下是安装记录~

1.windows安装Faiss-gpu版本

在windows上只是功能验证,怎么简单怎么来。只需要conda即可直接安装。注意:cuda版本一定要符合自己环境的cuda版本一致~

conda activate 你要用的环境
conda install -c conda-forge faiss-gpu cudatoolkit=11.8

2.Ubuntu安装Faiss-gpu版本

原来打算用源码构建,但是技术不足,踩雷不少屎坑,于是放弃直接用pip。简单又好用~~
cuda版本为12.x时

pip install faiss-gpu-cu12

cuda版本为11.x时

pip install faiss-gpu-cu11
;