horovod-yarn:Horovod on Yarn测试用例

时间:2024-04-24 05:32:49
【文件属性】:

文件名称:horovod-yarn:Horovod on Yarn测试用例

文件大小:4KB

文件格式:ZIP

更新时间:2024-04-24 05:32:49

Python

Horovod纱线 提前在纱线上提供​​一个horovod的方案。此方案可以在本地进行多进程的测试 driver启动一个rendevous服务器 各个工人通过注入一些horovod运行时变量,即可启动 司机 python3 driver.py --num_proc=2 任务 python3 tensorflow2_minist.py --port=34824 --rank=0 --size=2 --local_rank=0 --local_size=2 --cross_rank=0 --cross_size=1 python3 tensorflow2_minist.py --port=34824 --rank=1 --size=2 --local_rank=1 --local_size=2 --cross_rank=0 --cross_size=1


【文件预览】:
horovod-yarn-master
----tensorflow2_minist.py(5KB)
----driver.py(1KB)
----README.md(553B)

网友评论