当前位置:   article > 正文

faiss安装教程

faiss安装

faiss安装

使用Anaconda安装

使用Anaconda安装使用faiss是最方便快速的方式,facebook会及时推出faiss的新版本conda安装包,在conda安装时会自行安装所需的libgcc, mkl, numpy模块。
faiss的cpu版本目前仅支持Linux和MacOS操作系统,gpu版本提供可在Linux操作系统下用CUDA8.0/CUDA9.0/CUDA9.1编译的版本。
注意,上面语句中的cuda90并不会执行安装CUDA的操作,需要提前自行安装。

In [ ]:

  1. #安装cpu版本
  2. #更新conda
  3. conda update conda
  4. #先安装mkl
  5. conda install mkl
  6. #安装faiss-cpu
  7. conda install faiss-cpu -c pytorch
  8. #测试安装是否成功
  9. python -c "import faiss"

In [ ]:

  1. #安装gpu版本
  2. #确保已经安装了CUDA,否则会自动安装cup版本。
  3. conda install faiss-gpu -c pytorch # 默认 For CUDA8.0
  4. conda install faiss-gpu cuda90 -c pytorch # For CUDA9.0
  5. conda install faiss-gpu cuda91 -c pytorch # For CUDA9.1

注意,上面语句中的cuda90/91并不会执行安装CUDA的操作,需要提前自行安装。

编译安装

编译安装需要3个步骤:
1.编译C++文件;
2.编译Python界面;
3.编译GPU部分。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/羊村懒王/article/detail/732536
推荐阅读
相关标签
  

闽ICP备14008679号