Clean up the log info in running tf

__init__.py

try:
    # import tf 2.X
    import tensorflow.compat.v2 as tf
except Exception:
    # import tf 1.X
    import tensorflow as tf
    tf.logging.set_verbosity(tf.logging.ERROR)
print('Using TF {}'.format(tf.__version__))

# set logging level for tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

# set warning level for numpy
import warnings
warnings.simplefilter(action='ignore', category=FutureWarning)
warnings.simplefilter(action='ignore', category=DeprecationWarning)

import numpy as np

print(tf.test.is_gpu_available())