Skip to content

Commit f64dc3c

Browse files
Michael Norrisfacebook-github-bot
authored andcommitted
Update INSTALL.md to remove some raft references, add missing dependency (facebookresearch#4176)
Summary: Pull Request resolved: facebookresearch#4176 For issue: facebookresearch#4175 Reproduce on CI by: 1. Update build-pull-request.yml and action.yml to only run `conda install -c pytorch -c nvidia -c rapidsai -c conda-forge faiss-gpu-cuvs`, and also start a tmate session. 2. Log into the host 3. `cd` to ~/miniconda3/bin. Try to import faiss. It fails ``` runner@fv-az802-878:~/miniconda3/bin$ ./python Python 3.12.8 | packaged by conda-forge | (main, Dec 5 2024, 14:24:40) [GCC 13.3.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import faiss Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/home/runner/miniconda3/lib/python3.12/site-packages/faiss/__init__.py", line 17, in <module> from .loader import * File "/home/runner/miniconda3/lib/python3.12/site-packages/faiss/loader.py", line 149, in <module> from .swigfaiss import * File "/home/runner/miniconda3/lib/python3.12/site-packages/faiss/swigfaiss.py", line 13, in <module> from . import _swigfaiss ImportError: libnvJitLink.so.12: cannot open shared object file: No such file or directory >>> ``` Reviewed By: asadoughi Differential Revision: D69278685 fbshipit-source-id: 44c476c9ab3de5a43edb6da45cbb32a80df90160
1 parent dd5453c commit f64dc3c

2 files changed

Lines changed: 5 additions & 4 deletions

File tree

INSTALL.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ pre-release nightly builds.
66

77
- The CPU-only faiss-cpu conda package is currently available on Linux (x86-64 and aarch64), OSX (arm64 only), and Windows (x86-64)
88
- faiss-gpu, containing both CPU and GPU indices, is available on Linux (x86-64 only) for CUDA 11.4 and 12.1
9-
- faiss-gpu-raft [^1] package containing GPU indices provided by [NVIDIA RAFT](https://github.com/rapidsai/raft/) version 24.06, is available on Linux (x86-64 only) for CUDA 11.8 and 12.4.
9+
- faiss-gpu-cuvs [^1] package containing GPU indices provided by [NVIDIA cuVS](https://github.com/rapidsai/cuvs/) version 24.12, is available on Linux (x86-64 only) for CUDA 11.8 and 12.4.
1010

1111
To install the latest stable release:
1212

@@ -17,15 +17,15 @@ $ conda install -c pytorch faiss-cpu=1.10.0
1717
# GPU(+CPU) version
1818
$ conda install -c pytorch -c nvidia faiss-gpu=1.10.0
1919

20-
# GPU(+CPU) version with NVIDIA RAFT
21-
$ conda install -c pytorch -c nvidia -c rapidsai -c conda-forge faiss-gpu-raft=1.10.0
20+
# GPU(+CPU) version with NVIDIA cuVS
21+
$ conda install -c pytorch -c nvidia -c rapidsai -c conda-forge libnvjitlink faiss-gpu-cuvs=1.10.0
2222

2323
# GPU(+CPU) version using AMD ROCm not yet available
2424
```
2525

2626
For faiss-gpu, the nvidia channel is required for CUDA, which is not published in the main anaconda channel.
2727

28-
For faiss-gpu-raft, the rapidsai, conda-forge and nvidia channels are required.
28+
For faiss-gpu-cuvs, the rapidsai, conda-forge and nvidia channels are required.
2929

3030
Nightly pre-release packages can be installed as follows:
3131

conda/faiss-gpu-cuvs/meta.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ outputs:
7676
- libcublas {{ libcublas_constraints }}
7777
- libcuvs =24.12
7878
- cuda-version {{ cuda_constraints }}
79+
- libnvjitlink
7980
test:
8081
requires:
8182
- conda-build

0 commit comments

Comments
 (0)