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在系统库中找不到“ libmkldnn.so”。因此，您需要安装Intel-mkl-dnn。您有两种选择来解决它。

您可以自己编译并安装它。
或者，如果使用conda，则只需在conda云中使用二进制软件包。

conda install -c intel mkl-dnn

我安装没有MKL的pycaffe时的ImportError

1 个答案: