姓名:崔少杰 学号:16040510021
转载自:http://www.jianshu.com/p/fdc44bb41a39=有修改
【嵌牛导读】:基于DC/OS建立分布式机器学习系统
【嵌牛鼻子】:scheduler、server、worker
【嵌牛提问】:如何快速高效的建立分布式机器学习系统?
【嵌牛正文】:之前基于ps-lite实现了word2vec。下一步就是让这个算法能够分布式的跑起来。最简单的分布式方案大概是如下几步:把二进制文件copy到你要跑的机器上。
把每个进程需要读的数据文件copy到相应的地方。
当然,这一切都可以用一个脚本基于SSH来实现。但是会遇到几个问题:
你要跑的机器上有没有足够的CPU和内存资源。
你和别人的程序共同依赖了同一个动态库的不同版本。
你跑的过程中,你的进程被别人不小心kill掉了。
原来启动scheduler的机器IP换了,woker和server都要修改他们依赖的scheduler的地址。
新增加了一批机器,可以跑更多的进程了。
每次都要copy数据文件,每个进程读不同的文件。
......
为了解决上面的问题,你需要3个东西:
分布式的操作系统,来管理每台机器的CPU和内存。
分布式文件系统解决文件的存储和读写。
虚拟机系统来解决不同项目的不同版本的依赖 (当然你可以通过给每个程序都做一个单独的依赖库的目录,但如果依赖特别复杂,甚至是系统的某个底层依赖不同,就比较麻烦)。
而DCOS就是为了解决1,3而设计的。他是mesos+marathon+docker(其实DCOS是一个基于mesos的分布式操作系统,marathon,docker只是他的组件)。
mesos负责分布式的CPU和内存管理。
marathon是基于mesos的封装,提供rest API来使用mesos管理进程。
docker是一个虚拟机系统来保证不同人的程序所依赖的版本不会冲突。
下面详细介绍如何将ps-lite跑在DCOS上。ps-lite有3种类型的服务:
worker: 负责读取训练数据,更新参数。
server:负责存储参数,worker本地的参数会定时和server同步。
scheduler:负责服务发现,worker和server通过scheduler才知道各自在哪里。同时也负责worker和server的进程同步。比如要同时开始工作。只有一台。他的IP和端口事先需要知道,并通过环境变量在启动时告诉worker和server。
因为同一台宿主机上可能会启动多个worker和server。所以worker和server工作的端口不能事先指定,只能随机使用没有被占用的端口。
之前很多集成ps-lite的分布式系统时,都只解决了worker和server的分布式问题(比如用yarn来调度worker和server),而scheduler是找了一台机器单独部署的,这样scheduler的IP/端口事先就能知道并告诉worker和server。但是我希望把scheduler也启动在分布式系统中,方法如下:
将scheduler的任务命名为 sch,并在marathon中启动。同时指定一个端口(比如10000)。并且让docker使用HOST网络模式(用HOST模式是因为性能比较好,而ps-lite又是个网络性能要求很高的应用),且绑定到10000端口。这时,利用mesos-dns, scheduler会自动被赋予一个 sch.marathon.mesos 的域名,然后通过环境变量告诉worker和server。
worker和server原来会随机找端口。需要改成启动时docker被分配随机可用端口,并通过环境变量传给worker/server,然后worker/server绑定该端口。marathon提供了$PORT的环境变量来完成这个任务。
搞定服务发现,剩下就是分布式文件系统了。最简单可以用moosefs,然后mount到每台mesos-slave的机器。
{
"id": "/sw2vscheduler",
"cmd": null,
"cpus": 1,
"mem": 128,
"disk": 0,
"instances": 1,
"container": {
"type": "DOCKER",
"volumes": [],
"docker": {
"image": "registry.aliyuncs.com/xlvector/sw2v",
"network": "HOST",
"privileged": false,
"parameters": [],
"forcePullImage": true
}
},
"env": {
"LD_LIBRARY_PATH": "/tmp/sw2v/ps-lite/deps/lib",
"DMLC_PS_ROOT_URI": "sw2vscheduler.marathon.mesos",
"DMLC_NUM_WORKER": "4",
"DMLC_ROLE": "scheduler",
"DMLC_NUM_SERVER": "1",
"DMLC_PS_ROOT_PORT": "18001"
},
"portDefinitions": [
{
"port": 18001,
"protocol": "tcp",
"labels": {}
}
]
}
{
"id": "/sw2vserver",
"cmd": null,
"cpus": 1,
"mem": 1200,
"disk": 0,
"instances": 1,
"container": {
"type": "DOCKER",
"volumes": [],
"docker": {
"image": "registry.aliyuncs.com/xlvector/sw2v",
"network": "HOST",
"privileged": false,
"parameters": [],
"forcePullImage": true
}
},
"env": {
"LD_LIBRARY_PATH": "/tmp/sw2v/ps-lite/deps/lib",
"DMLC_PS_ROOT_URI": "sw2vscheduler.marathon.mesos",
"DMLC_NUM_WORKER": "4",
"DMLC_ROLE": "server",
"DMLC_NUM_SERVER": "1",
"SW2V_DATA": "/mnt/dcos/text8",
"DMLC_PS_ROOT_PORT": "18001"
},
"portDefinitions": [
{
"port": 10004,
"protocol": "tcp",
"labels": {}
}
]
}
{
"id": "/sw2vworker",
"cmd": null,
"cpus": 1,
"mem": 128,
"disk": 0,
"instances": 4,
"container": {
"type": "DOCKER",
"volumes": [
{
"containerPath": "/mnt/dcos",
"hostPath": "/mnt/dcos",
"mode": "RO"
}
],
"docker": {
"image": "registry.aliyuncs.com/xlvector/sw2v",
"network": "HOST",
"privileged": false,
"parameters": [],
"forcePullImage": true
}
},
"env": {
"LD_LIBRARY_PATH": "/tmp/sw2v/ps-lite/deps/lib",
"DMLC_PS_ROOT_URI": "sw2vscheduler.marathon.mesos",
"DMLC_NUM_WORKER": "4",
"DMLC_ROLE": "worker",
"DMLC_NUM_SERVER": "1",
"SW2V_DATA": "/mnt/dcos/text8",
"DMLC_PS_ROOT_PORT": "18001"
},
"portDefinitions": [
{
"port": 10003,
"protocol": "tcp",
"labels": {}
}
]
}