配置Tensorflow以使用所有CPU

Question

阅读: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,还是需要明确配置？

Answer 1

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

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