如何隐藏或解决有关 TF_FORCE_GPU_ALLOW_GROWTH 的 Tensorflow 警告？答案

【问题标题】：How to hide or solve Tensorflow warning about TF_FORCE_GPU_ALLOW_GROWTH?如何隐藏或解决有关 TF_FORCE_GPU_ALLOW_GROWTH 的 Tensorflow 警告？
【发布时间】：2021-04-01 15:46:40
【问题描述】：

运行使用了 TensorFlow 的程序时，会打印以下警告：

W tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.cc:39] 覆盖 allow_growth 设置，因为设置了 TF_FORCE_GPU_ALLOW_GROWTH 环境变量。原始配置值为 0。

我正试图摆脱这个警告。原因是因为我从另一个 Python 文件运行这个程序，所以现在警告显示在返回到该文件的结果中。我已经尝试了几件事来隐藏或解决警告，但是它们只导致将此警告的打印次数从 4 减少到 1。我现在正在做的是包括 'os.environ['TF_CPP_MIN_LOG_LEVEL' ] = '3'' （这应该不会导致打印警告或错误），以及包括 'os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = "true"' （也尝试了 'True' 和 '1'），以及导入 'warnings' 并将所有代码包装在 'with warnings.catch_warnings():' 中，并包括 'warnings.simplefilter("ignore" )'。

我很确定我的其余代码工作正常，所以我只想摆脱这个警告，或者以合适的方式解决它。

【问题讨论】：

只是为了确保。您确定在任何 TensorFlow 命令之前执行了 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3" 吗？因为对我来说，当我把它放在文件的开头时，它删除了所有关于“覆盖允许增长”的警告。

标签： python tensorflow gpu

【解决方案1】：

要消除 Tensorflow 警告，请在您的主脚本中包含这段代码。

import tensorflow as tf
tf.logging.set_verbosity(tf.logging.ERROR)

【讨论】：