diff --git a/docs/extras/integrations/vectorstores/faiss.ipynb b/docs/extras/integrations/vectorstores/faiss.ipynb index 13a5c07fec..b2f75811ae 100644 --- a/docs/extras/integrations/vectorstores/faiss.ipynb +++ b/docs/extras/integrations/vectorstores/faiss.ipynb @@ -23,9 +23,9 @@ }, "outputs": [], "source": [ - "#!pip install faiss\n", + "!pip install faiss-gpu # For CUDA 7.5+ Supported GPU's.\n", "# OR\n", - "!pip install faiss-cpu" + "!pip install faiss-cpu # For CPU Installation" ] }, { diff --git a/libs/langchain/langchain/vectorstores/faiss.py b/libs/langchain/langchain/vectorstores/faiss.py index 6a23fc8ae8..31930beaff 100644 --- a/libs/langchain/langchain/vectorstores/faiss.py +++ b/libs/langchain/langchain/vectorstores/faiss.py @@ -40,7 +40,7 @@ def dependable_faiss_import(no_avx2: Optional[bool] = None) -> Any: except ImportError: raise ImportError( "Could not import faiss python package. " - "Please install it with `pip install faiss` " + "Please install it with `pip install faiss-gpu` (for CUDA supported GPU) " "or `pip install faiss-cpu` (depending on Python version)." ) return faiss