首页 文章

sklearn和Tensorflow配备双CPU机器

提问于
浏览
1

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

我需要CPU计算能力用于sklearn和网格搜索 . Sklearn在2 cpu上的工作方式与在1上相同吗?当n_jobs = -1时,它会使用两个CPU上的所有核心吗?

在我的GPU上进行训练时,tensorflow是否只能在一个CPU上运行?如果我只是在TF网站上复制并粘贴MNIST for experts教程,那么它是否会使用CPU和我的GPU而不指定设备?

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

1 回答

  • 0

    从我读到的内容即使您添加了类似的指令

    with tf.Session() as sess:
      with tf.device("/cpu:0"):
        ...
    

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

相关问题