keras version to use with tensorflow-gpu 1.4

前端 未结 4 1365
没有蜡笔的小新
没有蜡笔的小新 2021-02-08 04:22

I am using ubuntu 16, with python 3, tf-GPU with keras.

I downgraded to tf 1.4 due to cuda errors as explained here

But now I am getting this error

4条回答
  •  一向
    一向 (楼主)
    2021-02-08 04:31

    If you are using keras exclusively with the tensorflow backend, I would recommend to use the keras implementation found in tf.keras rather than the keras module. That way, you won't scratch your head about possible incompatibilities or bugs (see also that question).

提交回复
热议问题