首页 > 解决方案 > keras和tensorflow升级后google colab GPU处理变得非常慢

问题描述

我已经使用以下代码升级了我的 tensorflow 和 keras:

!pip install tf-nightly-gpu-2.0-preview

现在模型学习的每个时期都需要 22 分钟,即升级前的 17 秒!!!我确实降级了 tensorflo 和 keras,但没有帮助!

标签: pythontensorflowkerasgoogle-colaboratory

解决方案


您可以使用 Runtime -> Reset all runtimes... 菜单项重置后端。(这比 快得多kill -9 -1,这需要一些时间来重新连接。)


推荐阅读