动态增长代码:
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)
动态增长代码:
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)
本文标题:服务器容易因gpu显存占满而卡死,那就尽量使用动态增长,牺牲一点
本文链接:https://www.haomeiwen.com/subject/yuwjuctx.html
网友评论