配置Tensorflow以使用所有CPU

时间:2016-09-08 15:42:35

标签: tensorflow

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

  

TensorFlow是否使用了所有可用的设备(GPU和CPU)   在我的机器上?

     

TensorFlow支持多个GPU和CPU。查看方法   有关如何使用带有TensorFlow的GPU的文档   TensorFlow为设备和CIFAR-10教程分配操作   对于使用多个GPU的示例模型。

     

请注意,TensorFlow仅使用具有计算功能的GPU设备   大于3.5。

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

1 个答案:

答案 0 :(得分:20)

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

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