tensorflow如何使用多个cpu

Question

如何在TensorFlow（python API）中为一个操作手动分配多个设备？

例如，为了将一个cpu分配给图形，我会这样做：

<select multiple>
  <option value="1">1</option>
  <option value="2">2</option>
  <option value="3">3</option>
</select>

如果我想将cpu 0,1,2,3 all分配给同一个op怎么办？

似乎我不能使用设备名称列表作为tf.device的arg。任何帮助或指导将不胜感激。谢谢！

Answer 1

你可以这样multiple with contexts：

with tf.device("/cpu:0") as cpu0, tf.device("/cpu:1") as cpu1:
    pass