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

【问题标题】：keras version to use with tensorflow-gpu 1.4与 tensorflow-gpu 1.4 一起使用的 keras 版本
【发布时间】：2018-08-15 09:56:43
【问题描述】：

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

由于here 解释的 cuda 错误，我降级到 tf 1.4

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

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

似乎这是张量流中的API change，新的keras不适合旧的tf。

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

【问题讨论】：

太好了，你能把这个作为答案吗？
现在发布。我最初没有把它作为答案，因为它并不能完全解决您关于哪个版本“正确”的问题，但我想如果有人有更好的信息，他们可以稍后发布。

标签： python tensorflow keras gpu version

【解决方案1】：

Keras - Tensorflow 版本兼容性是我自己多次遇到的常见问题。我将这个compatibility table 保存在我的书签中，匹配tensorflow 和keras 版本。 keras 2.0.8 似乎与 tensorflow 1.4 兼容。

【讨论】：

【解决方案2】：

如果您只在 tensorflow 后端使用 keras，我建议您使用 tf.keras 中的 keras 实现，而不是 keras 模块。这样一来，您就不会因为可能存在的不兼容或错误而摸不着头脑（另请参阅该问题）。

【讨论】：

【解决方案3】：

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

看releases page，好像version 2.0.8应该兼容TensorFlow 1.4；它已经大约一岁了，但至少你有一个起点。

【讨论】：

【解决方案4】：

我能够使用conda package manager 安装 keras 和 keras-gpu，并使用兼容的 tensorflow 和 cuda 版本来通过您的 TypeError: softmax()... 当我尝试使用加载 original BERT tensorflow checkpoint 时出现错误消息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

【讨论】：

【解决方案5】：

我使用的是 tensorflow 版本 1.15.4，安装 Keras 版本 2.3.1 解决了我的问题。

顺便说一句，this page 很有用。找到您想要的 tensorflow 版本，然后点击相应的Packages and 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

【讨论】：