繁体   English   中英

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

[英]How to change Keras/tensorflow version in Google colab?

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

问题是当我运行!pip install q keras==1.2.2 ,内核显示安装了 keras 1.2.2 但是当我使用keras.__version_检查它时它显示 2.1.6 。 tensorflow 也是同样的情况。

我该如何解决?

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

在这里你可以找到重启:

在此处输入图片说明

您的命令!pip install q keras==1.2.2 for 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 训练了我的模型。

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

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

" 要求您重新启动运行时及其按钮。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM