Google Colab - 您的会话因未知原因崩溃

Question

您的会话因未知原因崩溃

当我在 Google Colab 中运行以下单元格时：

from keras import backend as K
if 'tensorflow' == K.backend():
  import tensorflow as tf
  from keras.backend.tensorflow_backend import set_session
  config = tf.ConfigProto()
  config.gpu_options.allow_growth = True
  config.gpu_options.visible_device_list = "0"
  set_session(tf.Session(config=config))

我收到此消息是因为我已将两个数据集上传到谷歌驱动器。

有谁知道这条消息并可以给我一些建议吗？非常感谢您的每一个提示。

更新： 我总是收到消息

更新 我已从 Google Drive 中删除了数据集，但会话仍然崩溃。

Answer 1

Google Colab 正在崩溃，因为您正在尝试运行与GPUasRuntime相关的代码CPU。

如果将Runtime改为GPU则执行成功。下面提到了相同的步骤：

运行时 -> 更改运行时 -> GPU（从下拉列表中选择）。

请在Github Gist中找到工作代码。