Tensorflow зависает при попытке создать сеанс в Docker-контейнере

Question

Я пытаюсь развернуть веб-сайт для демонстрации с использованием CNN. Для этой цели я создал образ докера с зависимостями (в моем случае - tenorflow, keras и любые другие).

Мне удалось построить образ докера. Однако это не удается, когда я тестировал на некоторых образцах изображений. Я обнаружил, что проблема в tenorflow не может создать сеанс. Я все еще могу импортировать tenorflow и другие пакеты. Даже я могу запустить код для построения графа вычислений с кодом тензорного потока

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

Но он зависает всякий раз, когда я пытаюсь создать сеанс тензорного потока (Ex, sess = tf.Session()). Ниже приведен код для создания образа докера.

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)

Ожидаемый результат - увидеть сообщение «напечатано печатью» («Я надеюсь увидеть это сообщение ------------------------», c) ».

Answer 1

Столкнулся с подобной проблемой, когда я использовал Docker. Не удается найти ссылку на сообщение в блоге, но в основном рекомендуется установить глобальную переменную для GRAPH.

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

Вот руководство о том, как использовать графики и сессии. Вот руководство по TF, обслуживающему с Docker.