faiss简单测试方法

先把仓库克隆到本地,我这边还需要改cmake环境,在project上面加
set(CMAKE_CUDA_COMPILER /usr/local/cuda-11.8/bin/nvcc)
构建
mkdir build
cmake -B build .
编译,只需要编译faiss这部分就可以,(主目录下有很多测试代码,编译很慢,只编译faiss会快很多)
cd build
make -j faiss
这时候可以运行tutorial中的测试文件试试
还是在build文件夹内
make -j 2-IVFFlat
./tutorial/cpp/2-IVFFlat
运行结果为

同样的,可以自己编写测试代码进行测试。

然后如果修改了faiss的代码,需要重新编译运行

make -j 2-IVFFlat

./tutorial/cpp/2-IVFFlat

 

 这就实现了在本地修改测试faiss,可以输出faiss的运行中间结果以及执行时间

posted @ 2024-04-28 09:48  Echo宝贝儿  阅读(41)  评论(0编辑  收藏  举报