Tensorflow: 分叉后会话卡住了

版本：每晚为带有 GPU 的 Python2 预构建（刚刚）

我期待以下代码打印“10.0”3 次，但 session.run 卡在所有分叉进程中。

import tensorflow as tf
import multiprocessing as mp
import os

class Worker(mp.Process):
    def __init__(self, gid):
        self.gid = gid
        super(Worker, self).__init__()

    def run(self):
        G = tf.Graph()
        with G.as_default():
            x = tf.placeholder(tf.float32, shape=[])
            y = x * 2
            sess = tf.Session()
            print sess.run(y, feed_dict={x: 5})

G = tf.Graph()
with G.as_default():
    sess = tf.Session()
    with sess.as_default():
        x = tf.placeholder(tf.float32, shape=[])
        y = x * 2
        print sess.run(y, feed_dict={x: 5})

procs = [Worker(k) for k in range(2)]
for p in procs: p.start()
for p in procs: p.join()

删除主进程中的图形/会话将解决问题。 所以好像一旦有会话，我们就不能使用 fork 了？
无论有没有 GPU，问题都存在。

注意：此代码不会正常终止。 在 master 退出后，您可能需要手动终止分叉进程。