python - 禁用 TensorFlow 调试信息

标签 python tensorflow

我所说的调试信息是指 TensorFlow 在我的终端中显示的有关已加载库和找到的设备等的信息,而不是 Python 错误。

I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcurand.so locally
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:900] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
I tensorflow/core/common_runtime/gpu/gpu_init.cc:102] Found device 0 with properties: 
name: Graphics Device
major: 5 minor: 2 memoryClockRate (GHz) 1.0885
pciBusID 0000:04:00.0
Total memory: 12.00GiB
Free memory: 11.83GiB
I tensorflow/core/common_runtime/gpu/gpu_init.cc:126] DMA: 0 
I tensorflow/core/common_runtime/gpu/gpu_init.cc:136] 0:   Y 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:717] Creating TensorFlow device (/gpu:0) -> (device: 0, name: Graphics Device, pci bus id: 0000:04:00.0)
I tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.cc:51] Creating bin of max chunk size 1.0KiB
...

最佳答案

您可以使用 os.environ 禁用所有调试日志:

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 
import tensorflow as tf

在 tf 0.12 和 1.0 上测试

具体来说,

0 = all messages are logged (default behavior)
1 = INFO messages are not printed
2 = INFO and WARNING messages are not printed
3 = INFO, WARNING, and ERROR messages are not printed

关于python - 禁用 TensorFlow 调试信息,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35911252/

相关文章:

python - 在 chrome NativeHost 消息传递中使用 C exe 作为 Nativehost 应用程序而不是 python

python - 如何使用 Tensorflow API 重新采样音频文件

python - 如何为 TensorFlow 中的不同层或变量设置不同的学习率?

python - 黑盒功能的交互/交互

python - 如何在 Django 测试框架中修改 session

python - 将 YoloV3 输出转换为边界框、标签和置信度的坐标

python - tensorflow 中补丁的点积

tensorflow - 在遵循 tensorflow 的迁移学习示例时,添加 dropout 是否有助于减少过度拟合?

当父对象不从对象继承时,Python 2.x super __init__ 继承不起作用

python - 如何在 Windows 7 中运行 python 2 和 3?