导入keras的时候出现了版本不兼容的情况。
我的环境:

windows10
python3.5.6
tensorflow-gpu1.10.0

在使用pip install keras 默认版本安装完成后,使用

import keras

尝试导入keras出现异常:

>>> import keras
Using TensorFlow backend.
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "E:\program\Anaconda\Anaconda3\envs\tfgpuenv\lib\site-packages\keras\__init__.py", line 3, in <module>
    from . import utils
  File "E:\program\Anaconda\Anaconda3\envs\tfgpuenv\lib\site-packages\keras\utils\__init__.py", line 6, in <module>
    from . import conv_utils
  File "E:\program\Anaconda\Anaconda3\envs\tfgpuenv\lib\site-packages\keras\utils\conv_utils.py", line 9, in <module>
    from .. import backend as K
  File "E:\program\Anaconda\Anaconda3\envs\tfgpuenv\lib\site-packages\keras\backend\__init__.py", line 1, in <module>
    from .load_backend import epsilon
  File "E:\program\Anaconda\Anaconda3\envs\tfgpuenv\lib\site-packages\keras\backend\load_backend.py", line 90, in <module>
    from .tensorflow_backend import *
  File "E:\program\Anaconda\Anaconda3\envs\tfgpuenv\lib\site-packages\keras\backend\tensorflow_backend.py", line 54, in <module>
    get_graph = tf_keras_backend.get_graph
AttributeError: module 'tensorflow.python.keras.backend' has no attribute 'get_graph'

主要是因为tensorflow-gpu与keras的版本不兼容
参考版本兼容性信息网站之后(所列为该网站云服务器所提供的环境,因此表中的tensorflow与keras的搭配可以供参考,主要看Framework与Description两列):

https://docs.floydhub.com/guides/environments/

指定安装与tensorflow-gpu1.10.0相兼容的keras2.2.0:

pip install keras==2.2.0

此时再次导入keras就能成功进行,可顺便查看keras的版本:

>>> import keras
Using TensorFlow backend.
>>> keras.__version__
'2.2.0'

至此解决问题。

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐