是否可以使用CPU上的單個核心運行Tensorflow？

Question

有沒有辦法強制Tensorflow使用單個CPU內核而不是分佈式CPU內核？

我問，因爲在多個CPU上分配BLAS功能的回報通常是遞減的，至少在我嘗試過使用OpenBLAS和Julia的情況下。

我想要超參數搜索運行32個CPU核心，幾百個模型培訓。我認爲在單個CPU內核上並行訓練32個模型要比使用分佈式BLAS訓練32個模型系列要高效得多（我已經在Mocha Framework/Julia上展示了這些，這些變化很漂亮易於實現）。

Answer 1

您應該可以使用常規numctl --physcpubind以及with tf.device()。