首页 PaddleOCR 帖子详情
Out of memory error
收藏
快速回复
PaddleOCR 问答OCR 543 0
Out of memory error
收藏
快速回复
PaddleOCR 问答OCR 543 0

训练KIE中的SER训练时产生如下报错,batch_size与num_workers均已调成1仍显存不足,请问如何解决?

Out of memory error on GPU 0. Cannot allocate 937.335938PB memory on GPU 0, 3.078247GB memory has been allocated and available memory is only 7.823547GB.

Please check whether there is any other process using GPU 0.
1. If yes, please stop them, or start PaddlePaddle on another GPU.
2. If no, please decrease the batch size of your model.
(at ../paddle/fluid/memory/allocation/cuda_allocator.cc:86)

 

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