深度学习使用cuda加速,显示gpu out of memory的解决方法

1.问题的产生原因
我的原始代码中使用了下列config设置函数
config.gpu_options.allow_growth = True
导致显存占用不断增长,直到超出gpu显存

2.网上查询的解决方案
限制显存占用上限,如下列语句所示
config.gpu_options.per_process_gpu_memory_fraction = 0.7
结果:产生错误,代码无法运行

3.最终解决方案
不使用上述语句,注释掉他们
config = tf.ConfigProto()
# config.gpu_options.allow_growth = True
# config.gpu_options.per_process_gpu_memory_fraction = 0.7
config.allow_soft_placement = True
结果:代码成功运行

posted @ 2020-11-18 17:56  DreiHunde  阅读(1054)  评论(0)    收藏  举报