faiss库安装及问题解决!

前言

记录faiss库的安装过程。


一、faiss是什么?

faiss是一个Facebook AI团队开源的库,全称为Facebook AI Similarity Search,该开源库针对高维空间中的海量数据(稠密向量),提供了高效且可靠的相似性聚类和检索方法,可支持十亿级别向量的搜索,是目前最为成熟的近似近邻搜索库。

官方资源地址https://github.com/facebookresearch/faiss

二、安装步骤

1.cpu版本

conda install -c pytorch faiss-cpu

2.gpu版本

conda install -c pytorch faiss-gpu

三、问题解决

1.安装过程

faiss库安装时会自动下载很多包,也涉及一些下载较慢的。需要自己在网页上手动下载。下载后将其放到site-packages文件下就可以。安装过程中我下载了两个文件:

cudatoolkit-11.4.3-h39f8164_13.conda
libfaiss-1.7.4-cuda112hb18a002_0_cuda.conda

2.导入faiss库

安装完成后,直接在python中导入即可使用

import faiss

这里导入faiss库时还遇到了一些小问题,如下

ImportError: libmkl_intel_thread.so.1: cannot open shared object file: No such file or directory

这里需要进入安装faiss的虚拟环境下的lib文件夹中,将找不到的文件建立软连接即可。

ln -s libmkl_intel_thread.so   libmkl_intel_thread.so.1

前提时lib中存在libmkl_intel_thread.so文件
详情参考:https://zhuanlan.zhihu.com/p/589633827
有关软连接的知识参考:https://blog.csdn.net/jialibang/article/details/108247033


总结

解决了以上问题就可以正常使用faiss库啦,后续将用这个库实现文本数据库的相似度检索,也是一个全新未接触的领域,会继续分享相关知识,谢谢大家多多关注!

  • 15
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值