首页 Paddle Inference 帖子详情
使用CPU预测,开启MKLDNN时内存异常上涨 已解决
收藏
快速回复
Paddle Inference 问答推理 1337 1
使用CPU预测,开启MKLDNN时内存异常上涨 已解决
收藏
快速回复
Paddle Inference 问答推理 1337 1

使用CPU预测,开启MKLDNN时内存异常上涨

旧鸱c_
已解决
2# 回复于2021-08
请检查是否使用了变长输入,此时可以使用接口config.set_mkldnn_cache_capacity(capacity)接口,设置不同shape的缓存数,以防止缓存太多的优化信息占用较大的内存空间。
展开
0
收藏
回复
全部评论(1)
时间顺序
旧鸱c_
#2 回复于2021-08

请检查是否使用了变长输入,此时可以使用接口config.set_mkldnn_cache_capacity(capacity)接口,设置不同shape的缓存数,以防止缓存太多的优化信息占用较大的内存空间。

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