可以使用多cpu进程进行张量流动吗？

Question

我有一台有4个cpu的台式电脑。我是否可以在4个cpu上运行tensorflow，以便比单cpu更快？就像MPI程序一样？我可以使用tflearn来实现它吗？非常感谢！

Answer 1

是的，tensorflow将利用CPU上的多核。当您从源代码构建它时，您可以将其构建为仅在CPU上运行，也可以同时在CPU和GPU上运行。现在，我已经将tensorflow构建为在我的GPU上运行，但是如果我想将它放在CPU上，我会将以下代码放在我的python代码的顶部。

os.environ['CUDA_VISIBLE_DEVICES'] = ''

这只是设置一个环境变量，也可以通过终端来完成。当我在没有可用的cuda设备的情况下运行它时，它以100%的速度运行我的CPU的所有内核。我可以想象，如果你把它放在你的tflearn代码的顶端，它也会把它强制到cpu上。