深度学习(tensorflow)通用代码

概述

        在tensorflow的学习和做项目的过程中，总有一些环境设置类的代码需要编写，这类代码纯属复制粘贴，故而在本博客记录下来，适时可以复制粘贴，本文持续更新......

通用代码

Tensorflow日志级别设置

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
# 需要在导入tensorflow之前
import tensorflow as tf

值	日志级别	可输出的日志
0	INFO	INFO+WARNING+ERROR+FATAL
1	WARNING	WARNING+ERROR+FATAL
2	ERROR	ERROR+FATAL
3	FATAL	FATAL

 设置训练的GPU

TRAIN_GPU = [0, ]
os.environ["CUDA_VISIBLE_DEVICES"]  = ','.join(str(x) for x in TRAIN_GPU)
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)