使用Mindspore推理会出现内存泄漏问题

使用Mindspore推理会出现内存泄漏问题

在使用Mindspore训练完成模型以后:

  1. 设置使用GPU进行结果推理,在预测110张测试图片的预测时会出现内存占满爆炸;
  2. 设置使用CPU进行结果推理,在预测410张测试图片的预测时会出现内存占满爆炸;
  3. 预测5000张左右的数据图片使用的时间在3.5个小时以上,为什么推理的时间会这么久?

请问如何解决内存泄漏问题?以及为什么推理速度为什么这么慢

****************************************************解答*****************************************************

需要从以下内容定位问题:

1. 硬件环境:GPU/ASCEND/CPU

2. 软件环境:您目前的mindspore版本

3. 您出问题的代码,如果是model仓下的代码,请提供链接地址;如果是个人网络,请提供可还原问题的最小脚本及所需数据集。

4. 您完整的log日志。

你可能感兴趣的:(深度学习,人工智能,数据挖掘,python)