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的運行中間結果以及執行時間

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章