pip - 在 Google Colaboratory 上安装 faiss

Question

我尝试按照MUSE 项目的说明进行操作。

他们需要 PyTorch 和 Faiss。PyTorch 易于安装。但我发现安装Faiss有问题。

MUSE 上的说明告诉我使用

conda install faiss-cpu -c pytorch

但是 Google Colab 不支持conda（当我尝试!pip install conda时，它不起作用）

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

有没有办法安装 Faiss 或 conda？

Answer 1

这是我最终安装 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 2

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

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

代码来自这里：faiss/issues/890

Answer 3

试试这个方法！

1. 安装康达

   !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 /内容/

2. 安装 faiss

   ！是的，是的 conda install faiss-gpu -c pytorch