2020年3月24日
tensorflow nccl_op && 单机多卡 allreduce training
摘要： Tensorflow1.13.1中关于nccl的源码位于:tensorflow\core\kernels\nccl_ops.cc;tensorflow\core\nccl\nccl_manager.cc其中核心代码位于nccl_manager.cc文件，入口API为：AddToAllReduce 想 阅读全文
posted @ 2020-03-24 11:05 灰太狼锅锅 阅读(1583) 评论(0) 推荐(0) 编辑