Modifica del numero di fili in tensorflow su Cifar10

qualora si esegue il cifar10_eval.py, in crea 32 thread come segue:Modifica del numero di fili in tensorflow su Cifar10

I tensorflow/core/common_runtime/local_device.cc: 25] Dispositivo locale intra op parallelismo thread: 32

Penso che questo numero di thread è il numero di thread in esecuzione su CPU, ma quando controllo l'utilizzo, viene utilizzato solo il 400-500% delle CPU. Esiste comunque la possibilità di modificare questo numero di thread?

fonte

2015-12-21 Zk1001

Per configurare questo valore, è possibile passare un argomento tf.ConfigProto durante la costruzione della tf.Session:

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

fonte

2015-12-21 07:09:31 mrry

I fili sono pthreads standard che operano un pool di thread, creati dal codice [questo file] C++ (https: //github.com/tensorflow/tensorflow/blob/master/tensorflow/core/common_runtime/direct_session.cc). – mrry

@mrry Questo è utile. Puoi rispondere [this] (http://stackoverflow.com/questions/39774250/run-syntaxnet-on-multiple-cores) – kskp

Modifica del numero di fili in tensorflow su Cifar10

risposta

Problemi correlati