一种深度学习训练任务的工作方法及装置

摘要

本发明提供了一种深度学习训练任务的工作方法及装置，通过根据单服务器或多服务器节点中GPU剩余资源情况为多种深度学习训练任务类型分配GPU，达到了可以在保证GPU利用率的同时兼顾多种类深度学习训练任务的效果。其中，所述方法包括：获取用户输入的深度学习训练任务参数；从所述任务参数中确定所述深度学习训练任务类型，所述深度学习训练任务类型包括：单机型、多机型；根据不同的深度学习训练任务类型按照不同的策略进行GPU选取；根据所述GPU的位置选取与所述GPU之间通信距离最短的CPU进行工作。