存储器管理方法、人工智能处理系统及计算机程序产品

本发明提供一种存储器管理方法、人工智能处理系统及计算机程序产品，所述存储器管理方法，包括：获取人工智能模型的多个架构参数；根据多个架构参数，获取在人工智能模型的训练过程中所使用的存储器配置大小；根据存储器配置大小，于随机存取存储器中配置目标存储器区域，其中人工智能模型的训练过程中的多个训练处理阶段共同使用目标存储器区域；以及将对应人工智能模型的训练过程中的每个训练处理阶段的中继数据暂存在目标存储器区域，其中中继数据被传输于加速处理器模块、随机存取存储器与存储装置之间。由此，本发明可自动计算最适化的存储器配置大小，并通过共用存储器空间的设计，有效改善存储器碎片化的问题。