python - 当我在没有 MKL 的情况下安装 pycaffe 时出现导入错误

Question

我正在做一个项目，使用 IntelCaffe。但是当我运行我的 python 代码时:

import sys 
CAFFE_ROOT = '/path/to/caffe/'
sys.path.insert(0, CAFFE_ROOT + 'python')
import caffe
caffe.set_mode_cpu()

我收到以下错误:

File "caffemodel.py", line 6, in <module>
    import caffe
  File "../../caffe/python/caffe/__init__.py", line 37, in <module>
    from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver
  File "../../caffe/python/caffe/pycaffe.py", line 49, in <module>
    from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
ImportError: libmkldnn.so: cannot open shared object file: No such file or directory

我想知道怎么解决。希望有人帮助我。谢谢你!

Answer 1

如错误信息所示，Caffe在系统lib中没有找到“libmkldnn.so”。所以你需要的是安装Intel-mkl-dnn 。您有两种选择来解决它。

1. 您可以自行编译安装。
2. 或者如果您使用 conda，则仅使用 conda cloud 中的二进制包。

conda install -c intel mkl-dnn