Faiss利用mkl加速,構建索引訓練時出錯。

前言
記錄一下faiss構建索引訓練時碰到的一個坑。
Intel MKL FATAL ERROR: Cannot load libmkl_avx2.so or libmkl_def.so.
python

問題:

利用英特爾mkl(Math Kernel Library)庫加速faiss。 index.train()時報以下錯誤:git

Intel MKL FATAL ERROR: Cannot load libmkl_avx2.so or libmkl_def.so.

<!--more-->
github

解決方案

在調用faiss以前導入調用mkl。代碼以下:bash

import mkl
mkl.get_max_threads()

爲何這麼作,我還不太理解。猜想是conda安裝版本兼容的問題。具體能夠看我提的issue
code

補充

如出現mkl導入失敗的狀況。如索引

import mkl
ImportError: No module named mkl

解決方式以下:ip

### 執行:
$ conda install mkl
$ conda install mkl-service

轉自我的博客:https://kirio.vip/2019/03/28/...get

相關文章
相關標籤/搜索