显存不够怎么办?
skyjokerlin 发布于2020-07-24 浏览:2056 回复:8
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
个赞
共8条回复 最后由ticshen35回复于2020-11-22
#9ticshen35回复于2020-11-22

# GPU到这个路径下改这个
# C:\Users\admin\.paddlehub\modules\chinese_text_detection_db_server\module.py
# if use_gpu:
# config.enable_use_gpu(3000, 0) 8000改为3000就可以预测了 如果显存不够大,把这个值改为你的显存大小就可以

0
#8七年期限回复于2020-10-01

要不换个显卡 要不batchsize小一些吧

0
#7天然花生豆豆回复于2020-09-28

去tools/infoer/utility.py中把parser.add_argument("--gpu_mem", type=int, default=8000)的8000改为800就可以预测了

0
#6exemal回复于2020-08-22

为什么不使用百度大脑免费的V100做项目呢?

0
#5因为1224回复于2020-08-12

和你的一样。4G显存。。。。。。。。用超轻量模型也是这样。。

0
#4lstk1219回复于2020-08-09

遇到同样的问题,jetson nano

0
#3Randcase回复于2020-08-01

买个大点的显卡

0
#2ZengZhuoqi回复于2020-07-28

用超轻量的试试?

0
TOP
切换版块