如何使用Jupyter Notebook在多个CPU上运行Tensorflow

Question

我正在尝试使用多个CPU运行CNN模型，但不允许我指定要使用的CPU数量。

到目前为止我所做的

#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个选项中，它仅使用单个CPU。

顶部

如何在16个CPU上并行执行此作业？