目前可以通过 tf.device(...) 函数指定用于特定操作的 CPU 或 GPU,但是无论如何您都可以指定一个coreCPU 的?
目前没有用于将操作固定到特定核心的 API,尽管这将是一个很好的选择功能要求 https://github.com/tensorflow/tensorflow/issues。您可以通过创建多个 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)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)