install_faiss_gpu.sh

#!/bin/bash
set -ex

# Tested on 1xA100 machines on Lambda cloud

faiss_dir=${1:-thirdparty/faiss}

cd ${faiss_dir}

rm -rf build/
mkdir -p build/
cd build/

sudo apt-get update
sudo apt-get install -y libssl-dev libblas-dev libopenblas-dev liblapack-dev swig python3-numpy

## turn off -DFAISS_OPT_LEVEL=avx2 on grace hopper systems
cmake -DFAISS_ENABLE_GPU=ON \
      -DFAISS_ENABLE_PYTHON=OFF \
      -DBUILD_SHARED_LIBS=ON \
      -DFAISS_OPT_LEVEL=avx512 \
      -DBUILD_TESTING=OFF \
      -DCMAKE_BUILD_TYPE=Release \
     ..

sudo make -j$(nproc)
sudo make install