如何在 Google colab 中更改 Keras/tensorflow 版本?
Posted
技术标签:
【中文标题】如何在 Google colab 中更改 Keras/tensorflow 版本?【英文标题】:How to change Keras/tensorflow version in Google colab? 【发布时间】:2019-01-29 13:19:52 【问题描述】:我在 google colaboratory 上使用 keras/tensorflow,我需要回到以前的版本。
问题是当我运行 !pip install q keras==1.2.2
时,内核显示 keras 1.2.2 已安装,但是当我使用 keras.__version_
检查它时,它显示 2.1.6 。同样的情况也适用于张量流。
我该如何解决这个问题?
【问题讨论】:
【参考方案1】:我昨天遇到了这个问题。我很惊讶安装软件包没有效果。
但我意识到我需要重新启动内核。在 Colab 中,这称为Restart runtime
。重新启动后,您应该可以使用新版本。
在这里你可以找到重启:
【讨论】:
【参考方案2】:您对 Keras 的命令 !pip install q keras==1.2.2
没问题,只是您需要按照提到的其他答案重新启动运行时。
但对于 tensorflow,您应该使用 %tensorflow_version 1.x
.
正如他们所说的here:
避免对 GPU 和 TPU 使用 pip install 我们建议不要使用 pip install 为两个 GPU 指定特定的 TensorFlow 版本 和 TPU 后端。 Colab 从源代码构建 TensorFlow 以确保 与我们的加速器系列兼容。 TensorFlow 的版本 pip 从 PyPI 获取可能会遇到性能问题或可能 根本不工作。
正如它所说,当我尝试使用 pip 将 TensorFlow 降级到 1.5 时,它无法识别 GPU,只是使用 cpu 训练了我的模型。
【讨论】:
【参考方案3】:在最新的colab中,当你升级或降级一个模块时,在输出单元格中你会自动看到语句:
"警告:之前在此运行时中导入了以下包: [scipy] 您必须重新启动运行时才能使用新安装的版本。
" 要求您重新启动运行时及其按钮。
【讨论】:
以上是关于如何在 Google colab 中更改 Keras/tensorflow 版本?的主要内容,如果未能解决你的问题,请参考以下文章
在 Google colab 中更改目录(脱离 python 解释器)
在 google colab 中将 CUDNN 升级到 8.2
Google Colab 代码块,类似于 VScode 中的功能
如何在 google colab 中启用拼写检查器(colab 在 linux OS 上运行)?