确定AI模型的运行配置的方法、装置及相关设备

一种确定AI模型的运行配置的方法、装置及相关设备，涉及人工智能技术领域。获取AI模型中每个子模型的搜索空间，搜索空间包括多种运行配置及基于每种运行配置运行子模型时的运行耗时，运行配置包括计算资源的数量及并行策略；获取计算资源的最大数量；根据每个子模型的搜索空间以及最大数量，为每个子模型选择目标运行配置，多个子模型的目标运行配置构成目标配置组合，其中，多个子模型的目标运行配置中的计算资源的总量小于等于最大数量，各个目标运行配置对应的运行耗时的总和最小。如此，在计算资源受限的情况下自动搜索出各子模型分配到的计算资源数量及所采用的并行策略，能够有效降低人力成本，保障AI模型的整体训练效率达到较高水平。