首页 文章

配置Tensorflow以使用所有CPU

提问于
浏览
11

阅读:https://www.tensorflow.org/versions/r0.10/resources/faq.html它说:

TensorFlow是否使用我机器上可用的所有设备(GPU和CPU)? TensorFlow支持多个GPU和CPU . 有关TensorFlow如何为设备分配操作的详细信息,请参阅有关使用TensorFlow的GPU的操作方法文档,以及使用多个GPU的示例模型的CIFAR-10教程 . 请注意,TensorFlow仅使用计算能力大于3.5的GPU设备 .

这是否意味着Tensorflow可以自动使用给定计算机上的所有CPU,还是需要明确配置?

1 回答

  • 19

    CPU通过“设备”使用,它只是一个线程池 . 如果您需要更多线程,可以控制线程数:

    sess = tf.Session(config=tf.ConfigProto(
      intra_op_parallelism_threads=NUM_THREADS))
    

相关问题