OOM 超显存的解决办法

使用tensorflow训练模型的时候，有时候会报错OOM

import os
import tensorflow as tf

os.environ['CUDA_VISIBLE_DEVICES']='0,1'

# tf 1.13
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)

# tf 2.0
# gpu = tf.config.experimental.list_physical_devices('GPU')
# tf.config.experimental.set_memory_growth(gpu[0], True)
# gpu = tf.config.experimental.list_physical_devices('GPU')
# tf.config.experimental.set_memory_growth(gpu[1], True)
``