如何在Google Collab中更改Keras / tensorflow版本

She*_*Dev 4 python keras tensorflow google-colaboratory

我在google collab上使用keras / tensorflow,我需要返回到以前的版本。

问题是当我运行时!pip install q keras==1.2.2,内核显示已安装keras 1.2.2,但是当我使用keras.__version_它进行检查时,它显示了2.1.6。张量流也是如此。

我该如何解决?

blu*_*nox 5

我昨天有这个问题。我很惊讶安装软件包没有效果。但是后来我意识到我需要重新启动内核。在Colab中,这称为Restart runtime。重新启动后,您应该可以使用新版本。

在这里找到重启:

在此处输入图片说明


hos*_*ati 5

!pip install q keras==1.2.2对 Keras 的命令没问题,只是您需要像其他答案提到的那样重新启动运行时。
但是对于 tensorflow,你应该使用%tensorflow_version 1.x.
正如他们在这里所说:

避免对 GPU 和 TPU 使用 pip install 我们建议不要使用 pip install 为 GPU 和 TPU 后端指定特定的 TensorFlow 版本。Colab 从源代码构建 TensorFlow,以确保与我们的加速器组兼容。通过 pip 从 PyPI 获取的 TensorFlow 版本可能会遇到性能问题或根本无法工作。

正如它所说,当我尝试用 pip 将我的 TensorFlow 降级到 1.5 时,它无法识别 GPU,只是用 cpu 训练了我的模型。