Change faiss dependency from GPU to CPU

GPU not yet supported for Python 3.11 facebookresearch/faiss#2861 (comment) Signed-off-by: Ehud-Karavani <ehud.karavani@ibm.com>
BiomedSciAI · Oct 25, 2023 · 598b8fb · 598b8fb
1 parent d40b615
commit 598b8fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/causallib/contrib/requirements.txt b/causallib/contrib/requirements.txt
@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/cpu/  # To support cpu torch installation
 torch>=1.2.0
-faiss-gpu~=1.7.0
+faiss-cpu~=1.7.0  # Can also use gpu for some Python versions