当前位置 博文首页 > 无限迭代中......:PaddleOCR——申请显存不足【Allocate too mu

    无限迭代中......:PaddleOCR——申请显存不足【Allocate too mu

    作者:[db:作者] 时间:2021-07-19 22:30

    问题描述

    E0724 02:41:05.436789 3160 analysis_predictor.cc:515] Allocate too much memory for the GPU memory pool, assigned 8000 MB

    E0724 02:41:05.436789 3160 analysis_predictor.cc:518] Try to shink the value by setting AnalysisConfig::EnableGpu(...)

    问题分析

    PaddleOCR默认申请8G显存。

    venv/Lib/site-packages/paddleocr/paddleocr.py

    解决方案

    res = paddleocr.PaddleOCR(gpu_mem=1000).ocr(img=r'test.png')
    

    参考文章

    显存不够怎么办?

    win10系统下python使用GPU推理可能存在显存爆炸的情况

    cs