文件名称: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)