Faiss利用mkl加速,构建索引训练时出错。

58次阅读

共计 461 个字符,预计需要花费 2 分钟才能阅读完成。

前言记录一下 faiss 构建索引训练时碰到的一个坑。Intel MKL FATAL ERROR: Cannot load libmkl_avx2.so or libmkl_def.so.
问题:
利用英特尔 mkl(Math Kernel Library) 库加速 faiss。index.train() 时报如下错误:
Intel MKL FATAL ERROR: Cannot load libmkl_avx2.so or libmkl_def.so.
<!–more–>
解决方案
在调用 faiss 之前导入调用 mkl。代码如下:
import mkl
mkl.get_max_threads()
为什么这么做,我还不太理解。猜测是 conda 安装版本兼容的问题。具体可以看我提的 issue
补充
如出现 mkl 导入失败的情况。如
import mkl
ImportError: No module named mkl
解决方式如下:
### 执行:
$ conda install mkl
$ conda install mkl-service
转自个人博客:https://kirio.vip/2019/03/28/…

正文完
 0