編譯Faiss-gpu C++

基本介紹

使用Faiss版本【1.7.4】

該項目依賴于BLAS 組件 OpenBLAS 和 IntelMKL BLAS 【官方支持】

IntelMKL 會比 OpenBLAS 快的多。 【來自官方結論】

本機環(huán)境

Cuda ：11.1

Cuda-Driver： 515

InterMKL： 2021.2.0

Faiss ：1.7.4

注意：faiss僅適用于 CPU 的faiss-cpuconda 軟件包目前可在 Linux、OSX 和 Windows 上使用。包含faiss-gpuCPU 和 GPU 索引，可在 Linux 系統(tǒng)上使用，適用于各種版本的 CUDA?！緛碜怨俜健?/p>

1. 安裝InterMKL

使用apt安裝

# 添加密鑰
wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB \ | gpg --dearmor | sudo tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null# 添加源
echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | sudo tee /etc/apt/sources.list.d/oneAPI.list# 更新源
sudo apt update# 查找想要安裝的版本 
apt list -a intel-basekit# 安裝InterMKL 2021版本
apt-get install intel-basekit=2021.2.0-2883  # 配置MKL-ROOT 環(huán)境變量
export MKL_ROOT=/opt/intel/oneapi/mkl/2021.0# 配置InterMKL運行的環(huán)境變量
source /opt/intel/oneapi/setvars.sh intel64

2.執(zhí)行Cmake

執(zhí)行Cmake 命令，python 請將 FAISS_ENABLE_PYTHON改為ON

cmake -B build 
-DFAISS_ENABLE_GPU=ON 
-DFAISS_ENABLE_PYTHON=OFF