Advertisement

在docker中安装联邦学习框架:fedml

阅读量:

在docker中安装联邦学习fedml框架

这里主要介绍docker容器跨物理机的分布式fedml的安装方法

FedML项目的官方安装指南:http://doc.fedml.ai/#/installation-distributed-computing]

由于官方发布的安装教程并非基于Docker构建, 因此本人不得不在Docker环境中进行FedML的安装

几个关键的步骤是:

为部署fedml在多台物理节点上运行多台docker容器

为部署fedml在多台物理节点上运行多台docker容器

复制代码
    docker run --gpus all -itd --privileged --name=fedml_x -p 1000-1010:1000-1010 ufoym/deepo /bin/bash

此处分建议使用镜像ufoym/deepo配置。此处分建议使用镜像ufoym/deepo配置。该镜像已成功地配置了gpu环境,并整合了torch与tf框架。

要实现多个docker容器构成集群以便于它们之间能够相互发送数据。如果所有容器运行在同一台物理服务器上,则无需组建集群即可直接按照官方文档进行部署。
需要注意的一点是,在同一节点上的docker容器之间可以直接进行内网通信,并无需通过宿主机进行端口映射。
值得注意的是,在同一节点上的dock

全部评论 (0)

还没有任何评论哟~