tensorflow - 为什么在我使用 MKL 时 Tensorflow 会警告 AVX2？

Question

我正在使用带有 MKL 支持的 Tensorflow 的 Anaconda 发行版。

from tensorflow.python.framework import test_util
test_util.IsMklEnabled()

此代码打印 True .但是，当我编译我的 Keras 模型时，我仍然得到

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2

这不是我期望的行为。

Answer 1

tensorflow 执行的 MKL-DNN 部分(这是 MKL 工程师提供的主要加速)在运行时进行 JIT。因此，针对二进制创建的指令集与 MKL-DNN 代码无关。它只会影响其他数学，主要是 Eigen 库和任何没有被 MKL-DNN 函数替换的函数。