keras version to use with tensorflow-gpu 1.4
Solution 1
Keras - Tensorflow versions compatibility is a frequent problem that i have faced many times myself. I am keeping in my bookmarks this compatibility table, with matches of tensorflow
and keras
versions. It would seem that keras 2.0.8
is compatible with tensorflow 1.4
.
Solution 2
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).
Comments
-
thebeancounter about 3 years
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
TypeError: softmax() got an unexpected keyword argument 'axis'
Seems that this is an API change in tensorflow and new keras is not suitable for the old tf.
I can't find what is the correct keras version to use with tf 1.4 gpu. What is the correct one?