首页 PaddleOCR 帖子详情
paddleocr内存溢出问题
收藏
快速回复
PaddleOCR 问答OCR 2576 6
paddleocr内存溢出问题
收藏
快速回复
PaddleOCR 问答OCR 2576 6

内存溢出怎么解决?

0
收藏
回复
全部评论(6)
时间顺序
宇宙物语
#2 回复于2023-01

减少batchsize,利用checkpoint

0
回复
婚纱精锐神帅
#3 回复于2023-10
减少batchsize,利用checkpoint

请问checkpoint怎么利用呢,我把batchsize都降到1了还是会报错内存问题

0
回复
宇宙物语
#4 回复于2023-10
请问checkpoint怎么利用呢,我把batchsize都降到1了还是会报错内存问题[图片]

官方文档有解决方法。

#一旦不再使用即释放内存垃圾,=1.0 垃圾占用内存大小达到 10G 时,释放内存垃圾
export FLAGS_eager_delete_tensor_gb=0.0
#启用快速垃圾回收策略,不等待 cuda kernel 结束,直接释放显存
export FLAGS_fast_eager_deletion_mode=1
#该环境变量设置只占用 0%的显存
export FLAGS_fraction_of_gpu_memory_to_use=0

 

1
回复
婚纱精锐神帅
#5 回复于2023-10
官方文档有解决方法。 #一旦不再使用即释放内存垃圾,=1.0 垃圾占用内存大小达到 10G 时,释放内存垃圾 export FLAGS_eager_delete_tensor_gb=0.0 #启用快速垃圾回收策略,不等待 cuda kernel 结束,直接释放显存 export FLAGS_fast_eager_deletion_mode=1 #该环境变量设置只占用 0%的显存 export FLAGS_fraction_of_gpu_memory_to_use=0  
展开

谢谢答复,因为我一开始使用的是paddleocr-release-2.7,所以修改各种参数都不管用,应该是版本问题;后来我使用了paddleocr-release-2.6就直接搞定了。感谢!!

0
回复
宇宙物语
#6 回复于2023-10
谢谢答复,因为我一开始使用的是paddleocr-release-2.7,所以修改各种参数都不管用,应该是版本问题;后来我使用了paddleocr-release-2.6就直接搞定了。感谢!!

原来是这样,学习了。。。

0
回复
婚纱精锐神帅
#7 回复于2023-10
原来是这样,学习了。。。

你的方法我还没有试过,不过应该是可行的,对于内存溢出来说,所以还是很感谢的!!!后续我会试试

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户