Отображает ли TensorFlow все процессоры одной машины как ONE?

Question

Из экспериментов, которые я запускал, кажется, что TensorFlow автоматически использует все процессоры на одной машине. Кроме того, похоже, что TensorFlow относится ко всем ЦП в качестве/cpu: 0.

Я прав, что только разные графические процессоры одной машины индексируются и рассматриваются как отдельные устройства, но все процессоры на одной машине рассматриваются как одно устройство?

Есть ли способ, которым машина может иметь несколько процессоров, просматривающих ее с точки зрения TensorFlows?

Answer 1

По умолчанию все процессоры, доступные для процесса, агрегируются под устройством cpu:0.

Там в ответ на mrry here, показывающий, как создать логические устройства, такие как /cpu:1, /cpu:2

не кажется Там работать функциональность приколоть логических устройств с конкретными физическими ядрами или быть в состоянии использовать NUMA узлы в tensorflow.

Возможным обходным является использование распределенного TensorFlow с несколькими процессами на одной машине и использовать taskset на Linux, чтобы закрепить конкретные процессы для конкретных ядер