问是否可以使用CPU上的单个内核来运行Tensorflow？
EN

Stack Overflow用户

提问于 2016-11-09 23:06:36

回答 1查看 601关注 0票数 2

有没有办法迫使Tensorflow使用单个CPU核心而不是分布式CPU核心？

我之所以这样问是因为通常情况下，在多个CPU之间分配BLAS函数的回报会减少，至少在我尝试过使用OpenBLAS和Julia的情况下是这样。

我想要一个超参数搜索运行超过32个CPU核心，几百个模型培训。我希望在单个CPU核上并行地训练32个模型比使用分布式BLAS来训练32个模型要有效得多(我已经在Mocha Framework / Julia上演示了这一点，在那里这些变化很容易实现)。

回答已采纳

发布于 2016-11-10 04:01:37

您应该能够使用常规的numctl --physcpubind和with tf.device()。

票数 2

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/40517515

复制

相似问题

问是否可以使用CPU上的单个内核来运行Tensorflow？EN