paddleocr内存溢出问题
收藏
内存溢出怎么解决?
0
收藏
请登录后评论
减少batchsize,利用checkpoint
请问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
谢谢答复,因为我一开始使用的是paddleocr-release-2.7,所以修改各种参数都不管用,应该是版本问题;后来我使用了paddleocr-release-2.6就直接搞定了。感谢!!
原来是这样,学习了。。。
你的方法我还没有试过,不过应该是可行的,对于内存溢出来说,所以还是很感谢的!!!后续我会试试