我有一个Keras功能模型（带卷积层的神经网络），可以在张量流中很好地工作。我可以运行它，也可以安装它。
但是，使用tensorflow gpu时无法构建模型。
这是构建模型时抛出错误的代码行。
x = BatchNormalization(axis=1)(x)
错误消息为“ 批处理规范化形状必须为1级，但批处理规范化为4级”。
在我看来，这是Keras中的错误。如果是这样，那么如果有修复程序，如何将这个特定的修复程序安装到Windows / Anaconda环境中？
Keras版本是2.2.0和tensorflow gpu 1.8.0
有什么主意吗？
提前感谢大家。
答案 0 :(得分：3)
在我看来，您的计算机中存在tensorflow-gpu-keras版本不兼容。
尝试keras 2.1.6
for/nobreak