我有一个Keras功能模型（带卷积层的神经网络），它可以很好地处理张量流 . 我可以运行它，我可以适应它 .
但是，当使用tensorflow gpu时，无法构建模型 .
这是构建抛出错误的模型时的代码行 .
x = BatchNormalization(axis=1)(x)
错误消息是“ BatchNormalization Shape must be rank 1 but is rank 4 for batch_normalization ” .
在我看来，它是Keras的一个错误 . 如果有，如果有修复，我如何在Windows / Anaconda环境中安装此特定修复程序？
Keras版本是2.2.0和tensorflow gpu 1.8.0
任何的想法？
在此先感谢大家 .
1 回答
在我看来，你的机器中存在tensorflow-gpu-keras版本不兼容 .
尝试keras 2.1.6