本篇文章提供了了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