Bug 如何限制显存占用？

[TZJ12]

问题描述 / Problem Description

预留5个G给ocr模型使用，但是不够，仍然出现了爆显存的情况

运行环境 / Runtime Environment

nvidia linux

复现代码 / Reproduction Code

可能解决方案 / Possible solutions

[TZJ12]

paddlepaddle-gpu 2.6.2
rapidocr-paddle 1.3.25
python 3.10.12

[SWHL]

送入OCR的图像尺寸预先resize一下，应该可以缓解。