使用Mindspore推理会出现内存泄漏问题
在使用Mindspore训练完成模型以后:
- 设置使用GPU进行结果推理,在预测110张测试图片的预测时会出现内存占满爆炸;
- 设置使用CPU进行结果推理,在预测410张测试图片的预测时会出现内存占满爆炸;
- 预测5000张左右的数据图片使用的时间在3.5个小时以上,为什么推理的时间会这么久?
请问如何解决内存泄漏问题?以及为什么推理速度为什么这么慢
****************************************************解答*****************************************************
需要从以下内容定位问题:
1. 硬件环境:GPU/ASCEND/CPU
2. 软件环境:您目前的mindspore版本
3. 您出问题的代码,如果是model仓下的代码,请提供链接地址;如果是个人网络,请提供可还原问题的最小脚本及所需数据集。
4. 您完整的log日志。