一种显存优化方法、装置、计算机设备、存储介质及计算机程序产品

本申请公开一种显存优化方法、装置、计算机设备、存储介质及计算机程序产品，属于人工智能技术领域，该方法中，在多个芯片上分布式训练深度学习模型的过程中，当满足预设条件时，获取每个芯片的显存负载信息，以及每个芯片上多个对象的显存使用信息，多个对象对应这个芯片执行的不同训练任务，根据每个芯片的显存负载信息，以及这个芯片上多个对象的显存使用信息，更新这个芯片的对象集合，根据每个芯片的对象集合，向这个芯片发送显存优化指示，以触发这个芯片对这个对象集合中的对象应用显存优化技术。这样，每个芯片上应用显存优化技术的对象可随训练过程的变化而变化，显存优化与训练过程更匹配，显存优化更合理，因此，可提升模型训练效果。