与 tensorflow-gpu 1.4 一起使用的 keras 版本

新手上路,请多包涵

我正在使用 ubuntu 16、python 3、tf-GPU 和 keras。

由于 此处 解释的 cuda 错误,我降级到 tf 1.4

但是现在我收到了这个错误

TypeError: softmax() 得到了一个意外的关键字参数 ‘axis’

似乎这是 tensorflow 中的 API 更改,新的 keras 不适合旧的 tf.

我找不到与 tf 1.4 gpu 一起使用的正确 keras 版本。什么是正确的?

原文由 thebeancounter 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 396
1 个回答

Keras - Tensorflow 版本兼容性是我自己遇到过很多次的常见问题。我在我的书签中保留了这个 兼容性表,匹配了 tensorflowkeras 版本。似乎 keras 2.0.8tensorflow 1.4 兼容。

原文由 Ioannis Nasios 发布,翻译遵循 CC BY-SA 4.0 许可协议

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题