在 Google Colaboratory 上安装 faiss

Question

我尝试按照 MUSE project 上的说明进行操作。

他们需要 PyTorch 和 Faiss。 PyTorch 易于安装。但是我发现安装 Faiss.

有问题

MUSE 上的说明告诉我使用

conda install faiss-cpu -c pytorch

但是Google Colab 不支持conda（我试过!pip install conda，没用）

当我 !pip install faiss 时 Faiss 也没有工作。

有没有办法安装 Faiss 或 conda？

Answer 1

试试这个方法！

安装 conda

!wget -c -O anaconda.sh 'https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh'

!bash anaconda.sh -b

!cd /usr/bin/ && ln -sf /content/anaconda3/bin/conda conda

!cd /content/
安装 faiss

!是啊| conda 安装 faiss-gpu -c pytorch

Answer 2

下面是我最终安装 faiss 的方式。

!wget  https://anaconda.org/pytorch/faiss-cpu/1.2.1/download/linux-64/faiss-cpu-1.2.1-py36_cuda9.0.176_1.tar.bz2
!tar xvjf faiss-cpu-1.2.1-py36_cuda9.0.176_1.tar.bz2
!cp -r lib/python3.6/site-packages/* /usr/local/lib/python3.6/dist-packages/
!pip install mkl

然后，我就可以 import faiss 没问题了。警告是我没有使用 GPU。如果你想使用 GPU，你需要安装这个：

https://anaconda.org/pytorch/faiss-gpu/1.2.1/download/linux-64/faiss-gpu-1.2.1-py36_cuda9.0.176_1.tar.bz2

2020 年 6 月更新

正如@Kuffner 所说，您现在可以使用 !pip 来安装它。（我测试并简化了一点）

对于CPU

!apt install libomp-dev
!pip install faiss

对于 GPU

!pip install faiss-gpu

Answer 3

截至 2020 年 6 月，Colab GPU 运行时最简单的解决方案是：

!apt install libomp-dev
!python -m pip install --upgrade faiss faiss-gpu
import faiss

代码来自here: faiss/issues/890

在 Google Colaboratory 上安装 faiss

Installing faiss on Google Colaboratory

pip

conda

google-colaboratory

faiss

2020 年 6 月更新