一种面向神经网络模型部署的内存管理方法、装置及介质

本发明实施例公开了一种面向神经网络模型部署的内存管理方法、装置及介质；方法包括：按照需求对已训练的、待部署的深度学习模型进行解析，得到计算图；按照模式将其拆分为若干计算子图；选取内存管理策略，得到子图内存调度方案；对其进行整合，得到整体内存调度方案。实施本发明实施例所提供的内存管理方案，通过对计算图进行切分，划分为多个计算子图，通过对计算子图使用不同的内存管理策略得到局部最优的内存管理方案，增加了内存调度的搜索空间，进而得到比使用单一策略更优的、全局内存管理方案，减少了深度学习内存占用量，提高了芯片内存资源的利用率，使得用户可以在不改变芯片配置的前提下，部署更加复杂的深度学习模型。