Zmiana liczby wątków w TensorFlow na Cifar10

Ilekroć uruchomić cifar10_eval.py w tworzy 32 wątki w następujący sposób:Zmiana liczby wątków w TensorFlow na Cifar10

I tensorflow/Rdzeń/common_runtime/local_device.cc: 25] lokalnego urządzenia wewnątrz op równoległość thread: 32

Myślę, że ta liczba wątków to liczba wątków działających na procesorach, ale kiedy sprawdzam użycie, używa się tylko 400-500% procesorów. Czy mimo to można zmienić tę liczbę wątków?

Źródło

2015-12-21 Zk1001

Aby skonfigurować tę wartość, można zdać tf.ConfigProto argumentu przy konstruowaniu tf.Session:

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

Źródło

2015-12-21 07:09:31 mrry

Nitki są standardowe pthreads, które działają puli wątków, tworzone przez kod C++ w [archiwum] (https: //github.com/tensorflow/tensorflow/blob/master/tensorflow/core/common_runtime/direct_session.cc). – mrry

@mrry Jest to przydatne. Czy możesz odpowiedzieć [to] (http://stackoverflow.com/questions/39774250/run-syntaxnet-on-multiple-cores) – kskp

Zmiana liczby wątków w TensorFlow na Cifar10

Odpowiedz

Powiązane problemy