摘要
本公开提供了一种学习模型的硬件部署方法、相关装置和介质。该硬件部署方法包括:获取用于目标学习模型的处理单元部署的初始流水并行数和初始张量并行数;在部署方案节点跳转关系图上,定位初始流水并行数和初始张量并行数所在的初始部署方案节点;将部署方案节点跳转关系图和初始部署方案节点输入强化学习模型,得到使目标学习模型的推理性能最优化的目标部署方案节点,目标部署方案节点包括目标流水并行数和目标张量并行数;基于目标流水并行数和目标张量并行数,为目标学习模型部署处理单元。本公开实施例能扩大部署方法的适用范围,提高部署效率。