Faiss

From Alliance Doc
Revision as of 14:38, 25 April 2024 by Coulombc (talk | contribs) (Added page on Faiss)
(diff) ← Older revision | Latest revision (diff) | Newer revision → (diff)
Jump to navigation Jump to search
Other languages:

Faiss is a library for efficient similarity search and clustering of dense vectors. It contains algorithms that search in sets of vectors of any size, up to ones that possibly do not fit in RAM. It also contains supporting code for evaluation and parameter tuning. Faiss is written in C++ with complete wrappers for Python (versions 2 and 3). Some of the most useful algorithms are implemented on the GPU. It is developed primarily at Meta AI Research with help from external contributors.

Python bindings

The module contains bindings for multiple Python versions. To discover which are the compatible Python versions, run

Question.png
[name@server ~]$ module spider faiss/X.Y.Z

Or search directly faiss-cpu, by running

Question.png
[name@server ~]$ module spider faiss-cpu/X.Y.Z

where X.Y.Z represent the desired version.

Usage

1. Load the required modules.

Question.png
[name@server ~]$ module load StdEnv/2023 gcc cuda faiss/X.Y.Z python/3.11

where X.Y.Z represent the desired version.

2. Import Faiss

Question.png
[name@server ~]$ python -c "import faiss"

If the command displays nothing, the import was successful.

Available Python packages

Other Python packages depend on faiss-cpu or faiss-gpu bindings in order to be installed. faiss module provides:

  • faiss
  • faiss-gpu
  • faiss-cpu
Question.png
[name@server ~]$ pip list | fgrep faiss
faiss-gpu                          1.7.4
faiss-cpu                          1.7.4
faiss                              1.7.4

With the faiss module loaded, your package dependency for one of the Faiss named will be satisfied.