参考资料:https://www.tensorflow.org/tutorials/using_gpu
文中提到了两个控制方法:设置使用显存的百分比,以及设置逐渐增加使用的GPU显存。
关于第一种方法,可以参考Keras指定使用GPU、Keras设定GPU使用内存大小(Tensorflow backend)。
这里翻译第二种方法,代码如下:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)