使用multi-cpu进程可以张紧流量？

Question

我有一台带4个CPU的台式电脑。 我能用4个cpu运行tensorflow，所以它比单个cpu快吗？ 像MPI计划？ 我可以使用tflearn来实现它吗？ 非常感谢！

Answer 1

是tensorflow将利用CPU上的多个内核。从源代码构建它时，您可以将其构建为仅在CPU上运行或者能够在CPU和GPU上运行。现在我已经建立了在我的GPU上运行的tensorflow但是如果我想将它放在CPU上，我将以下内容放在我的python代码的顶部。

os.environ['CUDA_VISIBLE_DEVICES'] = ''

这只是设置一个环境变量，也可以通过终端完成。当我运行它没有cuda设备可用时，它运行我的CPU的所有核心100％。我想如果你把它放在你的tflearn代码的顶部，它也会强制它进入cpu。