如何在单进程模式下运行tensorflow？

Question

我正在尝试在不支持GPU的SLF 6上使用Python 2.7运行tensorflow 1.13.1。当我启动模型时，tensorflow似乎产生了多个子进程并并行运行我的模型，试图加载系统中的每个核心。虽然在大多数情况下这可能是我们想要的，但我的情况并非如此。我只想在单核上运行我的模型。

我尝试设置以下变量：

export OMP_NUM_THREADS=1
export KMP_BLOCKTIME=0
export KMP_AFFINITY=granularity=fine,verbose,compact,1,0

以不同的组合使用，但无法实现单核运行。

是否可以在“哑”单进程模式下运行Tensorflow？

Answer 1

在inter_op_parallelism_threads协议缓冲区中，有两个关于并行性intra_op_parallelism_threads和tf.ConfigProto的可配置选项。要使用单个过程，我想您可以尝试：

import tensorflow as tf
config = tf.ConfigProto(intra_op_parallelism_threads=1, 
                        inter_op_parallelism_threads=1, 
                        allow_soft_placement=True)

还有其他可能的并行形式，请参见mrry @的答案是thread。