跳转到主要内容

一个用于密集向量高效相似性搜索和聚类的库。

项目描述

faiss-wheels

Build PyPI

faiss python wheel包。

概述

此存储库提供用于构建faiss库轮包的脚本。

  • 使用cibuildwheel构建仅CPU版本。
  • 在Linux/Windows中捆绑OpenBLAS
  • 在macOS中使用Accelerate框架

还有一个源包,可以自定义构建过程。

注意 GPU包已支持至版本1.7.2,但自版本1.7.3起由于PyPI限制不再可用。

安装

通过以下方式安装二进制包

pip install faiss-cpu

构建源包

如果系统中已存在自定义构建的faiss库,构建源包以获得最佳性能。

先决条件

源包假定faiss已在系统中构建并安装。首先构建和安装faiss库。

cd faiss
cmake . -B build -DFAISS_ENABLE_GPU=OFF -DFAISS_ENABLE_PYTHON=OFF -DFAISS_OPT_LEVEL=avx512
cmake --build build --config Release -j
cmake --install build install
cd ..

请参阅官方faiss 安装说明了解如何构建和安装 faiss。

构建 sdist 需要 swig 3.0.12 或更高版本。

构建源代码发行版

以下示例构建并安装了支持 AVX512 的 faiss-cpu 源代码包。

export FAISS_OPT_LEVEL=avx512
pip install --no-binary :all: faiss-cpu

以下示例构建了一个 GPU 轮子。

export FAISS_ENABLE_GPU=ON
pip install --no-binary :all: faiss-cpu

有几个环境变量用于指定构建时的选项。

  • FAISS_INSTALL_PREFIX:指定 faiss 库的安装位置,默认为 /usr/local
  • FAISS_OPT_LEVEL:Faiss SIMD 优化,可以是 genericavx2avx512。注意,AVX 选项仅在 x86_64 架构中可用。
  • FAISS_ENABLE_GPU:将此变量设置为 ON 将构建 GPU 封装器。如果 faiss 与 GPU 支持一起构建,则设置此变量。
  • CUDA_HOME:指定构建 GPU 封装器时使用的 CUDA 安装位置,默认为 /usr/local/cuda

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源代码发行版

faiss_cpu-1.8.0.post1.tar.gz (63.7 kB 查看哈希值)

上传时间:

构建发行版

faiss_cpu-1.8.0.post1-cp312-cp312-win_amd64.whl (14.6 MB 查看哈希值)

上传时间: CPython 3.12 Windows x86-64

faiss_cpu-1.8.0.post1-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (27.0 MB 查看哈希值)

上传时间: CPython 3.12 manylinux: glibc 2.17+ x86-64

faiss_cpu-1.8.0.post1-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (3.6 MB 查看哈希值)

上传时间: CPython 3.12 manylinux: glibc 2.17+ ARM64

faiss_cpu-1.8.0.post1-cp312-cp312-macosx_11_0_arm64.whl (6.0 MB 查看哈希值)

上传时间: CPython 3.12 macOS 11.0+ ARM64

faiss_cpu-1.8.0.post1-cp312-cp312-macosx_10_14_x86_64.whl (7.4 MB 查看哈希值)

上传于 CPython 3.12 macOS 10.14+ x86-64

faiss_cpu-1.8.0.post1-cp311-cp311-win_amd64.whl (14.6 MB 查看哈希值)

上传于 CPython 3.11 Windows x86-64

faiss_cpu-1.8.0.post1-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (27.0 MB 查看哈希值)

上传于 CPython 3.11 manylinux: glibc 2.17+ x86-64

faiss_cpu-1.8.0.post1-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (3.6 MB 查看哈希值)

上传于 CPython 3.11 manylinux: glibc 2.17+ ARM64

faiss_cpu-1.8.0.post1-cp311-cp311-macosx_11_0_arm64.whl (6.0 MB 查看哈希值)

上传于 CPython 3.11 macOS 11.0+ ARM64

faiss_cpu-1.8.0.post1-cp311-cp311-macosx_10_14_x86_64.whl (7.3 MB 查看哈希值)

上传于 CPython 3.11 macOS 10.14+ x86-64

faiss_cpu-1.8.0.post1-cp310-cp310-win_amd64.whl (14.6 MB 查看哈希值)

上传于 CPython 3.10 Windows x86-64

faiss_cpu-1.8.0.post1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (27.0 MB 查看哈希值)

上传于 CPython 3.10 manylinux: glibc 2.17+ x86-64

faiss_cpu-1.8.0.post1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (3.6 MB 查看哈希值)

上传于 CPython 3.10 manylinux: glibc 2.17+ ARM64

faiss_cpu-1.8.0.post1-cp310-cp310-macosx_11_0_arm64.whl (6.0 MB 查看哈希值)

上传于 CPython 3.10 macOS 11.0+ ARM64

faiss_cpu-1.8.0.post1-cp310-cp310-macosx_10_14_x86_64.whl (7.3 MB 查看哈希值)

上传于 CPython 3.10 macOS 10.14+ x86-64

faiss_cpu-1.8.0.post1-cp39-cp39-win_amd64.whl (14.6 MB 查看哈希值)

上传于 CPython 3.9 Windows x86-64

faiss_cpu-1.8.0.post1-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (27.0 MB 查看哈希值)

上传于 CPython 3.9 manylinux: glibc 2.17+ x86-64

faiss_cpu-1.8.0.post1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (3.6 MB 查看哈希值)

上传于 CPython 3.9 manylinux: glibc 2.17+ ARM64

faiss_cpu-1.8.0.post1-cp39-cp39-macosx_11_0_arm64.whl (6.0 MB 查看哈希值)

上传于 CPython 3.9 macOS 11.0+ ARM64

faiss_cpu-1.8.0.post1-cp39-cp39-macosx_10_14_x86_64.whl (7.3 MB 查看哈希值)

上传于 CPython 3.9 macOS 10.14+ x86-64

faiss_cpu-1.8.0.post1-cp38-cp38-win_amd64.whl (14.6 MB 查看哈希值)

上传于 CPython 3.8 Windows x86-64

faiss_cpu-1.8.0.post1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (27.0 MB 查看哈希值)

上传于 CPython 3.8 manylinux: glibc 2.17+ x86-64

faiss_cpu-1.8.0.post1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (3.5 MB 查看哈希值)

上传于 CPython 3.8 manylinux: glibc 2.17+ ARM64

faiss_cpu-1.8.0.post1-cp38-cp38-macosx_11_0_arm64.whl (6.0 MB 查看哈希值)

上传于 CPython 3.8 macOS 11.0+ ARM64

faiss_cpu-1.8.0.post1-cp38-cp38-macosx_10_14_x86_64.whl (7.3 MB 查看哈希值)

上传时间 CPython 3.8 macOS 10.14+ x86-64

由以下支持