Using faiss from pytorch

rapidsai · Sep 25, 2023 · c430bb8 · c430bb8
1 parent 001c224
commit c430bb8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/conda/recipes/raft-ann-bench-cpu/meta.yaml b/conda/recipes/raft-ann-bench-cpu/meta.yaml
@@ -47,8 +47,7 @@ requirements:
   host:
     - glog {{ glog_version }}
     - matplotlib
-    - libfaiss-avx2 {{ faiss_version }} # [linux64]
-    - libfaiss {{ faiss_version }} # [aarch64]
+    - faiss-cpu {{ faiss_version }}
     - nlohmann_json {{ nlohmann_json_version }}
     - python
     - pyyaml
@@ -57,8 +56,7 @@ requirements:
   run:
     - glog {{ glog_version }}
     - h5py {{ h5py_version }}
-    - libfaiss-avx2 {{ faiss_version }} # [linux64]
-    - libfaiss {{ faiss_version }} # [aarch64]
+    - faiss-cpu {{ faiss_version }}
     - matplotlib
     - python
     - pyyaml

diff --git a/conda/recipes/raft-ann-bench/meta.yaml b/conda/recipes/raft-ann-bench/meta.yaml
@@ -73,8 +73,7 @@ requirements:
     # Temporarily ignore faiss benchmarks on CUDA 12 because packages do not exist yet
     {% if cuda_major == "11" %}
     - faiss-proc=*=cuda
-    - libfaiss {{ faiss_version }} # [aarch64]
-    - libfaiss-avx2 {{ faiss_version }} # [linux64]
+    - libfaiss {{ faiss_version }}
     {% endif %}
     - h5py {{ h5py_version }}
     - benchmark
@@ -96,8 +95,7 @@ requirements:
     # Temporarily ignore faiss benchmarks on CUDA 12 because packages do not exist yet
     {% if cuda_major == "11" %}
     - faiss-proc=*=cuda
-    - libfaiss {{ faiss_version }} # [aarch64]
-    - libfaiss-avx2 {{ faiss_version }} # [linux64]
+    - libfaiss {{ faiss_version }}
     {% endif %}
     - h5py {{ h5py_version }}
     - benchmark