sklearn和Tensorflow配备双CPU机器

时间:2016-08-24 14:15:04

标签: scikit-learn tensorflow cpu

我正在考虑构建一台用于机器学习的双CPU机器。我已经在我目前的装备中拥有一个快速的GPU,但我限制在32GB的DDR3,我有一个i7-4790k,我打算升级到双E5 2683 v3' s。

我需要CPU计算能力用于sklearn和网格搜索。 Sklearn是否在2 cpu上运行,与1对1相同?当n_jobs = -1?

时,它会使用两个CPU上的所有内核吗?

在我的GPU上训练时,tensorflow是否仅适用于一个CPU?如果我只是在TF网站上复制并粘贴MNIST专家教程,那么它会使用CPU和我的GPU而不指定设备吗?

我选择不把它放在超级用户论坛上,因为它更多的是关于软件而不是硬件。

1 个答案:

答案 0 :(得分:0)

即使你添加了像

这样的指令,我也会阅读
with tf.Session() as sess:
  with tf.device("/cpu:0"):
    ...

它将其视为建议,并且可以在认为合适时使用GPU。 我想它可能会使用其他CPU