tensorflow - 将 cpu 核心传递给 multiworkermirrored strategy()

Question

我想使用 cpu 实现镜像策略，但我不知道如何构建要传递给的参数mirroredstrategy()。这是gpu的代码行，distribution = tf.contrib.distribute.MultiworkerMirroredStrategy(["/device:GPU:0", "/device:GPU:1", "/device:GPU:2"])

我可以将“/device:GPU:0”更改为“/device:CPU:0”，但这似乎只使用一个核心，或者确实如此，我将如何检查？

标签： tensorflow

Answer 1

TensorFlow 可以开箱即用地使用多个 CPU 内核，因此在这种情况下您不需要使用策略。MultiworkerMirroredStrategy仅当您想使用多台机器进行训练时才需要。这些机器每个都可以有 GPU 或只有 CPU。