Не удается увидеть журналы Tensorflow в блокноте Google Colab

Question

Я играл с tenisflow apis на виртуальной записной книжке Google Colab.Я хочу увидеть отображение устройства моей виртуальной машины colab.

Как уже упоминалось в руководстве по разработке tenorflow, я могу установить флаг (log_device_placement = True), чтобы включить ведение журнала.https://www.tensorflow.org/guide/using_gpu

Ниже мой код работает на ноутбуке colab -

import tensorflow as tf
# Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)

tf.logging.set_verbosity(tf.logging.INFO)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print(sess.run(c))

Но, похоже, он не работает на ноутбуке colab.Однако он работает с локальной консолью терминала Jupyter для ноутбуков.

Есть идеи, как включить ведение журнала на платформе Google Colab?

Answer 1

Похоже на проблему TensorFlow: https://github.com/tensorflow/tensorflow/issues/3047

Или проблема с юпитером: https://github.com/ipython/ipython/issues/1230

Вот обходной путь использования сторонней библиотеки:

!pip install wurlitzer

import tensorflow as tf
# Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)

tf.logging.set_verbosity(tf.logging.INFO)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

# Runs the op.
from wurlitzer import pipes

with pipes() as (out, err):
  print(sess.run(c))

print (out.read())

Полная тетрадь: https://colab.research.google.com/drive/1Z5FVCD_z8EMmyd31PsjQffQV_K7dDLfj