尝试在Docker容器中创建会话时，Tensorflow冻结

Question

我正在尝试使用CNN部署用于演示的网站。 为了达到这个目的，我构建了一个具有依赖关系的docker镜像（在我的情况下为tensorflow，keras和其他杂项）。

我设法建立了docker镜像。但是，当我在一些示例图像上进行测试时，它失败了。我发现问题是tensorflow无法创建会话。我仍然可以导入tensorflow和其他软件包。甚至我都可以运行一个代码来使用张量流代码构建计算图

(Ex,  
a=tf.constant(7)
b=tf.constant(10)
c = tf.add(a,b)
).

但是，每当我尝试创建张量流会话(Ex, sess = tf.Session()).时，它就会冻结 下面是构建docker映像的代码。

def CNN_forward_example(self,image):

    a=tf.constant(7)
    b=tf.constant(10)
    c = tf.add(a,b)
    sess = tf.Session()
    print("I hope to see this message------------------------",c)

预期结果是看到消息“由print打印（“我希望看到此消息------------------------”，c） “。

Answer 1

在我使用Docker时遇到了类似的问题。找不到指向博客文章的链接，但基本上建议为GRAPH设置全局变量。

GRAPH = tf.Graph()
# and then where you need it
with GRAPH.as_default():
    sess = tf.Session()

这里是guide，介绍如何使用图表和会话。这是guide关于与Docker一起使用的TF。