faiss python安装_如何在ubuntu 18.04上安装faiss GPU

最近因为要做多路召回的相关研究需要使用faiss进行召回，很多同学困在了安装的流程上。所以想把如何安装流程做个小结。

最简单的方法就是通过Anaconda来进行安装，facebook官方会不定期的推送稳定版本到conda频道。

下面我讲介绍如何在刚安装完毕的Ubuntu 18.04上安装faiss GPU。

下面介绍如何安装CUDA 10.0到ubuntu.

安装CUDA 10.0

下载所需要的本地deb安装文件，大概1.6GB左右，等待下载结束就好了。

继续下面的操作指南

Installation Instructions:
`sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb`
`sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
`sudo apt-get update`
`sudo apt-get install cuda`

安装Conda

登录conda的网站

https://www.anaconda.com/products/individual

下载需要的conda版本

这里我选择的是Python 3.8和64-Bit (x86) Installer (529MB)

下载完后，执行

sudo sh Anaconda3-2020.11-Linux-x86_64.sh

conda的channel现在支持CUDA版本9.0，10.0的官方安装，如果你的CUDA版本是其他版本，也可以选择编译faiss源代码来代替官方版本。

使用conda官方的faiss版本来安装

请参考下面的命令来安装

# CPU version only
conda install faiss-cpu -c pytorch# GPU version
conda install faiss-gpu cudatoolkit=8.0 -c pytorch # For CUDA8
conda install faiss-gpu cudatoolkit=9.0 -c pytorch # For CUDA9
conda install faiss-gpu cudatoolkit=10.0 -c pytorch # For CUDA10

这样，你的faiss就安装好了。由于我的机器在用conda安装的时候碰到了网络问题，所以我不得不考虑使用源代码进行安装。

Nvidia 1080Ti显卡的使用方案

我的ubuntu是英伟达的1080Ti，我也想尝试使用源代码进行编译。于是下载下列代码

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

进入faiss目录

CPU-Only Build

cmake -B build . -DCUDAToolkit_ROOT=/usr/local/cuda-10.0/ -DFAISS_ENABLE_GPU=OFF -DPython_EXECUTABLE=/home/jiancong/anaconda3/bin/python3
make -C build -j16
cd build/faiss/python/ && python setup.py install

GPU Build

cmake -B build . -DCUDAToolkit_ROOT=/usr/local/cuda-10.0/ -DFAISS_ENABLE_GPU=ON -DPython_EXECUTABLE=/home/jiancong/anaconda3/bin/python3
make -C build -j16
cd build/faiss/python/ && python setup.py install

进入刚才生成的build目录，执行

make demo_ivfpq_indexing
cd demos
./demo_ivfpq_indexing

执行到这里如果一切ok，我们再测试python接口

python -c "import faiss, numpy
faiss.Kmeans(10, 20).train(numpy.random.rand(1000, 10).astype('float32'))"

还是没问题，很好，Faiss编译安装完毕，去嗨吧。

参考

^CUDA下载网站 https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=debnetwork