ubuntu中Keras报出omp警告

Question

环境：
ubuntu16.04 anaconda5.3.1
tensorflow1.13.1
keras 2.2.4

遇到的问题和下面这个链接相同，但我看不懂怎么调。。。求大神告知！！
https://github.com/tensorflow/tensorflow/issues/14496

Answer 1

以下回答参考 皆我百晓生、券券喵儿 等免费微信小程序相关内容作答，并由本人整理回复。

在您的Ubuntu环境中，您遇到了一个名为“OMP”的警告。这个警告通常表示OpenMP（多处理器并行）在处理TensorFlow中的某些操作时遇到了问题。它可能意味着您的系统没有足够的CPU核心来运行这些操作。

要解决这个问题，您可以尝试以下步骤：

1. 更新Python库：首先，确保您的Python库是最新的。您可以使用pip install --upgrade pip命令更新pypi库。然后，使用python -m ensurepip --default-pip命令安装确保pip。

2. 检查CPUs：打开终端或命令提示符，并运行以下命令以检查您的系统上有多少个CPU核心：

   lscpu
   

3. 执行优化：如果您发现系统中有足够的CPU核心，但仍然遇到问题，可以尝试执行以下优化操作之一：

   a. 更改KMP_AFFINITY参数：这将告诉OpenMP如何分配CPU核心。例如，使用以下命令更改KMP_AFFINITY设置：

    ```
    export OMP_NUM_THREADS=1
    ```
   

   b. 使用tf.cast函数：虽然这可能会提高性能，但在某些情况下可能导致错误，因为它强制TensorFlow模型将所有值都转换为浮点数。因此，请谨慎使用。

4. 检查是否有其他依赖项：有时，某些依赖项可能会影响OpenMP的性能。您可以尝试禁用某些库，看看是否有所改善：

   a. 从TensorFlow仓库卸载某些版本的库：

    ```
    git rm /tmp/tensorflow*
    ```
   

   b. 禁用某些特定的依赖项：

    ```
    pip uninstall <dependency_name>
    ```
   

请注意，上述建议针对的是特定情况下的解决方案，具体取决于您的工作负载和TensorFlow模型的具体需求。如果问题持续存在，您可能需要联系TensorFlow团队寻求更深入的帮助。