服务器容易因gpu显存占满而卡死，那就尽量使用动态增长，牺牲一点
2019-10-09 本文已影响0人
你说你要一场
动态增长代码：
import keras.backend.tensorflow_backend as KTF
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
KTF.set_session(sess)
