0%

服务器容易因gpu显存占满而卡死,那就尽量使用动态增长,牺牲一点性能也是可以的

动态增长代码:

1
2
3
4
5
6
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)