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

时间:2018-08-23 08:10:27

标签: python tensorflow keras google-colaboratory

我在google collab上使用keras / tensorflow,我需要返回到以前的版本。问题是当我运行!pip install q keras == 1.2.2 时,内核显示keras 1.2.2已安装,但是当我使用 keras .__ version _ 进行检查时,它却显示2.1.6。张量流也是如此。我该如何解决

3 个答案:

答案 0 :(得分:3)

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

在这里找到重启:

enter image description here

答案 1 :(得分:2)

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

避免在GPU和TPU上使用pip安装,建议不要使用 点安装以为两个GPU指定特定的TensorFlow版本 和TPU后端。 Colab从源头构建TensorFlow以确保 与我们的加速器车队兼容。 TensorFlow的版本 通过pip从PyPI获取的内容可能会遇到性能问题或可能 根本不起作用。

就像它说的那样,当我尝试使用pip将TensorFlow降级到1.5时,它无法识别GPU,而只是使用cpu训练了我的模型。

答案 2 :(得分:0)

在最新的colab中,当您升级或降级模块时,在输出单元格中您将自动看到以下语句:

“警告:先前在此运行时中导入了以下软件包:   [臭] 您必须重新启动运行时才能使用新安装的版本。

”,要求您重新启动运行时以及相应的按钮。