pip - 在 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

这是我最终安装 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安装它。 (我测试并简化了一 pip )

CPU用

!apt install libomp-dev
!pip install faiss

对于 GPU

!pip install faiss-gpu