Faiss源码安装

本篇文章提供了了Faiss安装的教程(CPU版安好了,GPU版有点问题,后补)


Step 1 下载FAISS

git clone https://github.com/facebookresearch/faiss

Step 2 生成配置

cd faiss
./configure --with-cuda=/usr/local/cuda-9.0 --prefix=/home/cver/software/faiss/ --with-python=python3  

configure生成makfile文件,这里需要配置cuda目录,安装目录prefix

Step 3 安装faiss

我运行时存在一些问题不能安装GPU版本,这里我把GPU部分屏蔽了,以后再解决。

vim makefile.inc
-------------------------------

NVCC         = /usr/local/cuda-9.0/bin/nvcc
CUDA_ROOT    = /usr/local/cuda-9.0
//CUDA_ARCH    = -gencode=arch=compute_35,code=compute_35 \
//-gencode=arch=compute_52,code=compute_52 \
//-gencode=arch=compute_60,code=compute_60 \
//-gencode=arch=compute_61,code=compute_61 \
//-gencode=arch=compute_70,code=compute_70 \
//-gencode=arch=compute_75,code=compute_75

修改之后,编译,安装

make
make install

Step 4 安装swing3

swing3是C++转python的桥梁…具体可以查一下。
首先要下载swing3

wget https://sourceforge.net/projects/swig/files/swig/swig-3.0.12/swig-3.0.12.tar.gz
tar -zxvf swig-3.0.12.tar.gz

然后配置config

bash ./configure --prefix=/home/cver/software/swing3 --without-pcre
make
make install

配置环境变量

vim ~/.bashrc

-----------------------添加以下两句
export SWIG_PATH=/home/cver/software/swing3/bin
export PATH=$SWIG_PATH:$PATH

Step 5 生成python包

回到faiss目录下

make py

仍然需要修改配置文件

cd python
vim Makefile

---------------------我没有sudo权限,所以我安装到自己目录下
$(PYTHON) setup.py install --prefix=/home/cver/.local

配置好后编译,安装

make install
发布了140 篇原创文章 · 获赞 26 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/CLOUD_J/article/details/100895267