Skip to main content

FAISS 1.14.0 with GPU + NVIDIA cuVS support — unofficial pip wheel (Linux x86_64, Ampere/SM86+)

Project description

FAISS 1.14.0 built from source with full GPU and NVIDIA cuVS support, and AVX2 CPU fallback.

This fills a gap that does not exist elsewhere:
- faiss-gpu-cu12 on PyPI (including 1.14.0): GPU only, no cuVS
- faiss-gpu-cuvs official: conda only, not on PyPI

| Property | Value |
|---|---|
| FAISS version | 1.14.0 |
| Python | 3.12 |
| CUDA | 12.x |
| GPU architecture | Ampere (SM86 native; SM89/90 via PTX JIT) |
| cuVS | Enabled (FAISS_ENABLE_CUVS=ON) |
| CPU SIMD | AVX2 |
| BLAS | OpenBLAS |
| Platform | Linux x86_64 |

GPU Architecture Compatibility:
- SM 86 (native): RTX 3080, RTX 3090, RTX 3090 Ti, A40
- SM 89, 90 (PTX JIT): RTX 4000 series, H100
- SM 80 and below (will NOT work): A100, RTX 2000 series, V100 and older

Installation:

# 1. Install RAPIDS dependencies (hosted on https://pypi.nvidia.com)
pip install libcuvs-cu12==25.10.0 librmm-cu12==25.10.0 libraft-cu12==25.10.0 \
rapids-logger "nvidia-nvjitlink-cu12>=12.9" --extra-index-url https://pypi.nvidia.com

# 2. Install system dependency
sudo apt-get install -y libopenblas-dev

# 3. Install this package
pip install faiss-gpu-cu12-cuvs --extra-index-url https://pypi.nvidia.com

Note: Conflicts with faiss-cpu and faiss-gpu-cu12. Do not install alongside them.
Note: RAPIDS dependencies (libcuvs-cu12, librmm-cu12, libraft-cu12, rapids-logger,
nvidia-nvjitlink-cu12) are hosted at https://pypi.nvidia.com and must be
available via that index when installing.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

faiss_gpu_cu12_cuvs-1.14.0-cp312-cp312-manylinux_2_38_x86_64.whl (42.4 MB view details)

Uploaded CPython 3.12manylinux: glibc 2.38+ x86-64

File details

Details for the file faiss_gpu_cu12_cuvs-1.14.0-cp312-cp312-manylinux_2_38_x86_64.whl.

File metadata

File hashes

Hashes for faiss_gpu_cu12_cuvs-1.14.0-cp312-cp312-manylinux_2_38_x86_64.whl
Algorithm Hash digest
SHA256 cebfd400e0b40264867de5f3a9327fdfd2c3eda0407ab5f0a8145c54438ebbfe
MD5 69c213c9027cce3efccaacc5a623e5e5
BLAKE2b-256 a50fb6f671902d11889ef05f9c6027076d2568dd821c8314fa816ac14f9b0d0c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page