是否可以使用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()。