1. 限制GPU的使用:
config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.5 config.gpu_options.allow_growth = True with tf.Session(config=config) as sess:
2. 每次迭代之后,清理掉得到的模型:
tf.reset_default_graph()