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

Question

我使用的是ubuntu 16,python 3,tf-GPU和keras.

由于这里解释的cuda错误,我降级到了1.4

但现在我收到了这个错误

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

似乎这是张量流中的API变化,而新的keras不适合旧的tf.

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

Answer 1

Keras-Tensorflow版本的兼容性是我多次面对的问题。我过去使用过（保留在书签中），此链接带有tensorflow和keras版本的匹配。我相信这keras 2.0.8与tensorflow 1.4

Answer 2

似乎没有关于哪个 Keras 版本针对哪个 TensorFlow 版本的正确文档。解决问题的最快方法可能就是一次降级 Keras 一个版本，直到找到一个可以工作的版本（或者，相反，从您知道可以工作的版本一次升级一个版本，直到它崩溃为止）。如果您觉得这很乏味，您可以将其作为二分搜索来完成。

查看releases页面，似乎2.0.8版本应该兼容TensorFlow 1.4；它已经有大约一年的历史了，但至少你已经有了一个起点。