Tensorflow:执行具有CPU特定核心的操作

Question

目前可以指定与特定操作的tf.device(...)函数一起使用哪个CPU或GPU,但是在哪里可以指定CPU 的核心？

Answer 1

目前没有用于将ops固定到特定核心的API,尽管这会产生很好的功能请求.您可以通过创建多个CPU设备来近似此功能,每个CPU设备都具有单线程线程池,但这不能保证保持核心固定解决方案的位置:

with tf.device("/cpu:4"):
  # ...

with tf.device("/cpu:7"):
  # ...

with tf.device("/cpu:0"):
  # ...

config = tf.ConfigProto(device_count={"CPU": 8},
                        inter_op_parallelism_threads=1,
                        intra_op_parallelism_threads=1)
sess = tf.Session(config=config)