Tensorflow: توقفت الجلسة بعد الشوكة

الإصدار: تم إنشاؤه ليليًا مسبقًا لـ Python2 w / GPU (الآن)

أتوقع أن يطبع الكود التالي "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()

ستؤدي إزالة الرسم البياني / الجلسة في العملية الرئيسية إلى حل المشكلة. لذلك يبدو أنه بمجرد وجود جلسة ، لا يمكننا استخدام الشوكة؟
توجد المشكلة مع GPU وبدونه.

ملاحظة: هذا الرمز لا ينتهي بشكل طبيعي. ستحتاج على الأرجح إلى إنهاء العمليات المتشعبة يدويًا بعد خروج السيد.