RuntimeError: CUDA error: out of memory

[bdus]

change:

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=1.0)
[via]

into :

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.9)
config = tf.ConfigProto(allow_soft_placement=True, gpu_options=gpu_options)
config.gpu_options.allow_growth = True

[shenGrant]

想问一下 这个问题解决了吗 我是运行demo的时候加载预训练参数 gpu 显存不够了

[guess3w]

@shenGrant 像上面这兄弟说的，将text_detect.py中gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.9)这个比例改小点，我的显示是TITAN V，改为了0.6，内存不溢出了。