在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)
还没有任何评论哟~
