2015-12-21 49 views
12

每當運行cifar10_eval.py,在創建32個線程如下:改變TensorFlow線程數量上Cifar10

我tensorflow /型芯/ common_runtime/local_device.cc:25]本地設備內運算的並行線程:32

我認爲這個線程數是在CPU上運行的線程數,但是當我檢查使用情況時,只有400-500%的CPU被使用。無論如何改變這個線程數量?

回答

22

要配置此值,則可以構建時傳遞tf.ConfigProto參數中tf.Session

NUM_THREADS = … 
sess = tf.Session(config=tf.ConfigProto(
    intra_op_parallelism_threads=NUM_THREADS)) 
+6

螺紋是操作一個線程池標準並行線程,由C++在[這個文件]代碼(HTTPS創建的: //github.com/tensorflow/tensorflow/blob/master/tensorflow/core/common_runtime/direct_session.cc)。 – mrry

+0

@mrry這很有用。你可以請回答[這](http://stackoverflow.com/questions/39774250/run-syntaxnet-on-multiple-cores) – kskp