一种用于深度学习框架分布式训练的方法和装置

摘要

本发明公开了一种用于深度学习框架分布式训练的方法。该方法包括：根据IP地址文件中存储的各个容器的IP地址，将在分布式训练中使用的各种类型的节点分别分配到各个容器中，并且配置每个容器的中的训练参数；采用在各个容器中的节点来运行分布式训练的相应的进程。通过上述方式，可以使用容器的方式来部署运行分布式训练的节点，然后通过设置容器的端口地址，可以在一台服务器中部署多个容器，从而在总体上减少了进行深度学习框架分布式训练所使用的服务器的总数，节约分布式训练所使用的服务器硬件资源，并且节约运行服务器所需的电力资源。还公开了对应的用于深度学习框架分布式训练的装置。