keras version to use with tensorflow-gpu 1.4

前端 未结 4 1425
没有蜡笔的小新
没有蜡笔的小新 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条回答
  •  萌比男神i
    2021-02-08 04:52

    There does not seem to be proper documentation on which Keras version targets which TensorFlow version. The quickest way to solve your problem may be just downgrading Keras one version at a time until you find one that works (or, conversely, upgrading one version at a time from one that you know to work until it breaks). If you find that tedious you can do it as a binary search.

    Looking at the releases page, it seems that version 2.0.8 should be compatible with TensorFlow 1.4; it's about a year old already but at least you have an starting point there.

提交回复
热议问题