服务器容易因gpu显存占满而卡死,那就尽量使用动态增长,牺牲一点性能也是可以的 发表于 2019-10-09 更新于 2020-05-07 阅读次数: 本文字数: 218 阅读时长 ≈ 1 分钟 动态增长代码:123456import keras.backend.tensorflow_backend as KTFimport tensorflow as tfconfig = tf.ConfigProto()config.gpu_options.allow_growth=True sess = tf.Session(config=config)KTF.set_session(sess)