ResourceExhaustedError 解决方案

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()

 

posted @ 2019-05-28 14:45  爱学英语的程序媛  阅读(1101)  评论(0编辑  收藏  举报