Configuring Tensorflow to use all CPU's

后端 未结 1 826
再見小時候
再見小時候 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)
提交回复
热议问题