Cannot install Faiss in Google Collab

🤗Datasets
1

Has anyone gotten FAISS index to work in Google Collab (free version)?

I tried following the example mentioned in the doc.

However it throws an error at add_faiss_index() method saying you must install Faiss to use FaissIndex. This is despite the fact that I do

!pip install faiss-cpu
!pip install faiss-gpu
2 Likes
2

You can find the solution to this issue in the Discord thread.

3

Were you able able to find a solution? I am encountering the same issue but in Snowflake. I have tried to follow the Discord threat but it seems it has been deleted.

4

I am getting the same problems. I do !pip install faiss-gpu and it gives me a message error. I ignore it and run the code and it actually works. Later I try to run again and it doesn’t and have this issue with faiss.

Trying to fix it but so far nothing

1 Like
5

I found the solution. The following code should fix it.

!apt install -y libomp-dev
!python -m pip install --upgrade faiss-gpu
2 Likes
6

I used this command and it works.

!pip install faiss-gpu-cu11==1.10.0
5 Likes
7 
!pip install faiss-gpu-cu12

This does work for me too. Depending on your cuda version you might need to install faiss-gpu-cu12. You can check your cuda version:

import torch
print(torch.version.cuda)

Another thing to note, I got an error when I imported datasets library and then installed faiss, maybe somethings are cached. It’s better to do the installation first and then import all your libraries.

1 Like