显存不够怎么办?
0
收藏
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(...)
E0724 02:41:05.910789 3160 analysis_predictor.cc:515] Allocate too much memory for the GPU memory pool, assigned 8000 MB
E0724 02:41:05.911761 3160 analysis_predictor.cc:518] Try to shink the value by setting AnalysisConfig::EnableGpu(...)
使用了paddleOCR，结果提示这个了，我的显存只有4GB，能否修改掉默认的GPU内存池设置?修改了一些环境变量也没用
收藏
点赞
0
个赞
请登录后评论
TOP
切换版块
试过了，这只是个提示，并不会影响识别过程。
当use_gpu=True时，对显卡初始化时默认会申请8GB显存。如果实际显存没有那么多，paddle就会给出这个提示，但仍然会正常完成显卡的初始化，并且只要模型不超显存，后续的识别仍然可被正常执行。
# GPU到这个路径下改这个
# C:\Users\admin\.paddlehub\modules\chinese_text_detection_db_server\module.py
# if use_gpu:
# config.enable_use_gpu(3000, 0) 8000改为3000就可以预测了 如果显存不够大，把这个值改为你的显存大小就可以
要不换个显卡 要不batchsize小一些吧
去tools/infoer/utility.py中把parser.add_argument("--gpu_mem", type=int, default=8000)的8000改为800就可以预测了
为什么不使用百度大脑免费的V100做项目呢？
和你的一样。4G显存。。。。。。。。用超轻量模型也是这样。。
遇到同样的问题，jetson nano
买个大点的显卡
用超轻量的试试？