python - 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

我尝试单击“重置运行时”以在启动时运行 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 团队的成员。)