我正在尝试在多个 GPU 上运行子类 Keras 模型。代码按预期运行,但是,在代码执行过程中会出现以下“警告”:
“2 个 IndexedSlices 不支持高效的 allreduce”
这是什么意思?我遵循了 Tensorflow 2.0 Beta 指南上的多 GPU 教程。我也在我的输入管道中使用数据集 API。
multi-gpu deep-learning keras tensorflow
deep-learning ×1
keras ×1
multi-gpu ×1
tensorflow ×1