阅读: 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,还是需要明确配置?
答案 0 :(得分:20)
CPU通过"设备使用"这只是一个线程池。如果您需要更多线程,可以控制线程数:
sess = tf.Session(config=tf.ConfigProto(
intra_op_parallelism_threads=NUM_THREADS))