Configuring Tensorflow to use all CPU's

后端 未结 1 822
再見小時候
再見小時候 2020-12-08 16:29

Reading : https://www.tensorflow.org/versions/r0.10/resources/faq.html it states :

Does TensorFlow make use of all the devices (GPUs and CPUs) avail

相关标签:
1条回答
  • 2020-12-08 17:08

    CPUs are used via a "device" which is just a threadpool. You can control the number of threads if you feel like you need more:

    sess = tf.Session(config=tf.ConfigProto(
      intra_op_parallelism_threads=NUM_THREADS))
    
    0 讨论(0)
提交回复
热议问题