我正在使用 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 许可协议
Keras - Tensorflow 版本兼容性是我自己遇到过很多次的常见问题。我在我的书签中保留了这个 兼容性表,匹配了
tensorflow
和keras
版本。似乎keras 2.0.8
与tensorflow 1.4
兼容。