version keras à utiliser avec tensorflow-gpu 1.4
J'utilise Ubuntu 16, avec Python 3, GPU avec keras.
J'ai rétrogradé à tf 1.4 en raison d'erreurs cuda comme expliqué ici
Mais maintenant je reçois cette erreur
TypeError: softmax () a un argument de mot clé inattendu 'axe'
On dirait qu'il s'agit d'un modification de l'API en tensorflow et que les nouvelles versions ne sont pas adaptées à l'ancienne fonction.
Je ne trouve pas quelle est la version correcte de keras à utiliser avec tf 1.4 gpu. Quel est le bon?
Keras - La compatibilité de la version Tensorflow est un problème que j'ai rencontré à plusieurs reprises. J'ai utilisé par le passé (conservé dans les signets), ce link , avec des correspondances de versions tensorflow et keras. Je crois que keras 2.0.8 est compatible avec tensorflow 1.4
Si vous utilisez keras exclusivement avec le backend tensorflow, je vous conseillerais d'utiliser l'implémentation de keras présente dans tf.keras plutôt que dans le module keras. De cette façon, vous ne vous laisserez pas ébranler par d'éventuelles incompatibilités ou bugs (voir aussi cette question).
Il ne semble pas exister de documentation appropriée sur la version de Keras qui cible la version de TensorFlow. Le moyen le plus rapide de résoudre votre problème consiste peut-être simplement à déclasser la version de Keras une par une jusqu'à ce que vous en trouviez une qui fonctionne (ou, inversement, à mettre à niveau une version à la fois d'une version que vous savez travailler jusqu'à ce qu'elle se casse). Si vous trouvez cela fastidieux, vous pouvez le faire comme une recherche binaire.
En regardant la page releases , il semble que la version 2.0.8 devrait être compatible avec TensorFlow 1.4; cela fait déjà environ un an, mais au moins vous avez un point de départ.