All Reduce算法汇总

1、朴素All Reduce

假设两两GPU间的网络带宽为1GB/s,梯度总量1GB,

参考链接

1)Baidu All Reduce:

http://andrew.gibiansky.com/

https://github.com/baidu-research/baidu-allreduce

2)Horovod:

https://arxiv.org/pdf/1802.05799.pdf

猜你喜欢

转载自www.cnblogs.com/yangwenhuan/p/11390823.html
ALL