Tensorflow：将会话传递给python多进程

如何解决Tensorflow：将会话传递给python多进程

我认为您不能像这样在两个进程之间共享“状态”tf.Session()。我认为每个过程都需要自己的会话。

解决方法

我正在使用tensorflow预处理一些大图像。我遇到了内存快速崩溃的问题。我转而在python中使用多处理，以便在需要时可以完全释放内存。

问题是，我正在使用python的多进程队列，由于某种原因，我无法将我的tensorflow会话从父进程传递给子进程。使用一些先进的调试技术（即每隔几行打印一些内容），我注意到python只是在我使用会话的行内空闲，它不会抛出错误消息。

我的代码如下所示：

def subprocess(some_image,sess,q):
    with sess.as_default():
        # ... use sess and q ...
        print "All good and well" #This is printed
        some_image.eval() #Nothing happens here in console
        print "Still all good and well" #This is not printed

if __name__ == '__main__':
    # ... some initial operations ...
    some_image = read_some_image()

    sess = tf.Session()

    q = Queue()
    q.put(something)
    p = Process(target=subprocess,args=(some_image,q))
    p.start()
    p.join()

可能是什么问题呢？非常感谢！