在Google合作实验室上安装faiss

Question

我尝试遵循有关MUSE项目的说明。

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

关于MUSE的说明告诉我使用

conda install faiss-cpu -c pytorch

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

当我!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

尝试这个解决方案，它应该有效！

!pip install faiss-cpu --no-cache