我有一台带4个CPU的台式电脑。 我能用4个cpu运行tensorflow,所以它比单个cpu快吗? 像MPI计划? 我可以使用tflearn来实现它吗? 非常感谢!
答案 0 :(得分:0)
是tensorflow将利用CPU上的多个内核。从源代码构建它时,您可以将其构建为仅在CPU上运行或者能够在CPU和GPU上运行。现在我已经建立了在我的GPU上运行的tensorflow但是如果我想将它放在CPU上,我将以下内容放在我的python代码的顶部。
os.environ['CUDA_VISIBLE_DEVICES'] = ''
这只是设置一个环境变量,也可以通过终端完成。当我运行它没有cuda设备可用时,它运行我的CPU的所有核心100%。我想如果你把它放在你的tflearn代码的顶部,它也会强制它进入cpu。