silence-tensorflow 1.0.0

Simple python package to shut up Tensorflow warnings and logs.

Simple python package to shut up Tensorflow warnings and logs.

How do I get it?

As usual, just install it from pip:

pip install silence_tensorflow

This package WILL NOT install Tensorflow with it, as often custom versions are needed for each setup.

How do I use it?

You only need to import the package before import Tensorflow:

import silence_tensorflow
import tensorflow as tf

...

# your code

How does this work under the hood?

This package will set the KMP_AFFINITY system variable to “noverbose”, TF_CPP_MIN_LOG_LEVEL to level 2 (only errors logged) and silence both FutureWarning and FutureWarning.

If you need a custom value for KMP_AFFINITY you should reset it after importing the package, as follows:

import os
backup = os.environ["KMP_AFFINITY"]
import silence_tensorflow
os.environ["KMP_AFFINITY"] = backup