c++ - 无法运行使用 MKL 编译的 TensorFlow

Question

我像这样使用 CUDA 和 MKL 在 Ubuntu 16.04 上编译了最新的 TensorFlow

bazel build --config=opt --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" --config=mkl --config=cuda //tensorflow/tools/pip_package:build_pip_package

现在，当我尝试运行它时，我收到一条错误消息，指出无法找到英特尔的一个库。我还发现其他人在安装不同的 DNN 框架时遇到了这个问题 https://github.com/PaddlePaddle/Paddle/issues/3213并找到了英特尔文档 https://software.intel.com/en-us/articles/intel-mkl-dnn-part-1-library-overview-and-installation这基本上是说，据我所知，当您按照该文档中的说明进行操作时，这些文件应该可用。我遵循了这些指示，一切似乎都有效，但实际上那些 libmklml_intel.so 和 libiomp5.so 文件没有添加到 /usr/本地/lib.

>>> import tensorflow

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/__init__.py", line 24, in <module>
    from tensorflow.python import *
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow.py", line 52, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow.py", line 41, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
ImportError: libmklml_intel.so: cannot open shared object file: No such file or directory

编辑:实际上，在我克隆 mkl-dnn git 并按照 https://software.intel.com/en-us/articles/intel-mkl-dnn-part-1-library-overview-and-installation 中的说明进行操作后，它们位于 /mkl-dnn/external/mklml_lnx_2018.0.20170720/lib 中

Answer 1

实际上，在我克隆了 mkl-dnn git 并按照 https://software.intel.com/en-us/articles/intel-mkl-dnn-part-1-library-overview-and-installation 中的说明操作之后，这些库位于 /mkl-dnn/external/mklml_lnx_2018.0.20170720/lib 中

所以我将它们复制到 /usr/local/lib 并将这些文件夹包含在 .bashrc 中作为

export LIBRARY_PATH=/usr/local/lib:$LIBRARY_PATH导出 LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

并重新加载.bashrc

source ~/.bashrc

现在 TensorFlow 可以工作了。