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

Question

我正在使用 ubuntu 16，带有 python 3，带有 keras 的 tf-GPU。

我降级到TF 1.4由于CUDA错误，说明这里

但现在我收到这个错误

类型错误：softmax() 得到了一个意外的关键字参数“轴”

似乎这是tensorflow中的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 后端一起使用，我建议您使用tf.keras的 keras 实现而不是keras模块。 这样，您就不会因为可能的不兼容或错误而挠头（另请参阅该问题）。

Answer 3

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

查看发布页面，似乎2.0.8 版本应该与 TensorFlow 1.4 兼容； 它已经大约一岁了，但至少你有一个起点。

Answer 4

我能够使用conda 包管理器来安装 keras 和 keras-gpu，并使用兼容的 tensorflow 和 cuda 版本来解决您的TypeError: softmax() ... 当我尝试使用加载原始 BERT tensorflow 检查点时的错误消息keras-bert包：

$ conda create -n bert python=3.6
$ conda activate bert
$ conda install keras==2.0.8
$ conda install keras-gpu==2.0.8
$ pip install keras-pos-embd==0.10.0
$ pip install keras-transformer==0.22.0

Answer 5

我正在使用 tensorflow 1.15.4 版并安装 Keras 2.3.1 版解决了我的问题。

顺便说一句，这个页面很有用。 找到您想要的 tensorflow 版本，然后单击相应的Packages 和 Nvidia Settings 。

setuptools==41.0.0
numpy
matplotlib
pandas
pydub
scipy
tensorflow==1.15.4
keras==2.3.1
python_speech_features
praat-parselmouth
pyquaternion