标签： tensorflow distributed grpc infiniband
我们有一个包含InfiniBand互连的集群。我知道Distributed TF使用gRPC来处理网络呼叫，但我想知道在使用gRPC时是否有任何特殊要求来获得InfiniBand连接的速度。
答案 0 :(得分：3)
我认为答案是肯定的。 技术上不是Tensorflow需要infiniband支持而是更多的gRPC层。
在github gRPC存储库中快速查询infiniband并获得零结果强化了我的假设：https://github.com/grpc?query=infiniband