Colab 上的 Tensorflow Eager Execution

Question

我试图在 Colab 笔记本上使用 Tensorflow 启用即时执行，但我收到一条错误消息：

import tensorflow as tf
import tensorflow.contrib.eager as tfe

tf.enable_eager_execution()

ValueError: tf.enable_eager_execution must be called at program startup

我试图点击 "reset runtime" 以便在启动时 运行 tf.enable_eager_execution 但错误再次出现。

打开其他笔记本没有帮助。

在出现此错误后，我可以在 Colab 中执行哪些操作以允许立即执行？

谢谢。

Answer 1

这是我们这边的一个错误，现已修复，原始代码片段应该可以再次使用。参见 Tensorflow eager mode dose not work when GPU is enabled in google colab · Issue #262 · googlecolab/colabtools。

（注意：我在 Google Colaboratory 团队。）