tensorflow(gpu调用以及警告忽略)

#忽略警告

import tensorflow as tf
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"    #只显示error信息
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)'


import warnings
warnings.filterwarnings("ignore")

#调用GPU

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 1"  

config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))

with tf.Session(config=config) as sess:
    # 初始化参数
    saver = tf.train.Saver(max_to_keep=10000)
    config.gpu_options.per_process_gpu_memory_fraction = 0.9
    sess.run(tf.global_variables_initializer())
    sess.run(tf.local_variables_initializer())