今天在学习TensorFlow中的Keras模型时,在keras官方文档中运行其中的例子,但是一开始就遇到了很神奇的错误

ModuleNotFoundError: No module named 'tensorflow.keras'

我感到奇怪,看到其他博客里提到的这个问题也不多,我想肯定不是什么大问题。

在一篇博客中我发现可以keras的path输出来看看路径是否出现了问题。随后发现tensorflow.keras 的路径相当奇怪

是在 类似与 lib/tensorflow/xxx/api/xxx/keras  的路径中。后面考虑是否是版本问题,后面在github一个issue下面找到了答案,原来是tensorflow的版本问题。

tensorflow.keras.__version__

类似的操作,在tensorflow 1.10.0 以上的版本才存在,而我的tensorflow版本是1.8的,通过

//由于我的是gpu的版本
pip install tensorflow-gpu==1.10

完美解决了问题。(环境为tensorflow1.10 + CUDA 9.0 + cuDNN 7.1 + Keras 2.1)

--------------------------------------------------------补-------------------------------------------------------------------

友情提示:在替换tensorflow版本之前,要注意自己的CUDA版本,否则会出现不兼容的现象。

另外,在替换版本之前,可以看看自己的keras路径,在引入包的时候,在tensorflow和keras之间加入python

from tensorflow import keras
from tensorflow.python.keras import layers

 

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐