Как запустить Tensorflow на нескольких процессорах с помощью ноутбука Jupyter - PullRequest
Новая
       7

Как запустить Tensorflow на нескольких процессорах с помощью ноутбука Jupyter

0 голосов
/ 01 октября 2019

Я пытаюсь запустить свою модель CNN, используя несколько процессоров, но это не позволяет мне указать количество процессоров, которые я хочу использовать.

Что я сделал до сих пор 

#option 1
#session_conf = tf.ConfigProto(
#      intra_op_parallelism_threads=8,
#      inter_op_parallelism_threads=8)
#session = tf.Session(config=session_conf)

#option 2
#config = tf.ConfigProto()
#config.intra_op_parallelism_threads = 8
#config.inter_op_parallelism_threads = 8
#session = tf.Session(config=config)

#option 3
#config = tf.ConfigProto(device_count={'CPU': 16})
#session = tf.Session(config=config)

#option 4
config = tf.ConfigProto(device_count={'GPU':0, 'CPU':16})
session = tf.Session(config=config)

session.run(tf.global_variables_initializer())

При всех вышеперечисленных 4 опциях используется только один процессор.

top enter image description here

Как распараллелить эту работу на 16 процессорах?

...