GPU TensorFlow 2.3 - InvalidArgumentError: l'assertion a échoué: [0] [Op: Assert] name: EagerVariableNameReuse

Question

Environnement:

Windows 8.1 64 bits Python 3.7.7 64 bits TensorFlow 2.3 Keras 2.4

CUDA: 10,1 CuDNN: 7

Erreur : InvalidArgumentError: assertion failed: [0] [Op:Assert] name: EagerVariableNameReuse.

Problème : erreur ci-dessus lors d'une simple déclaration model = Sequential()lors de l'exécution du GPU

Mais si j'exécute le script sans GPU (sans définir les variables d'environnement CUDA / CuDnn, cela fonctionne comme prévu.

Donc effectivement, j'ai un GPU mais je ne peux pas l'utiliser avec TensorFlow (2.3)

Quelqu'un peut-il aider?

Answer 1

D'après ce que j'ai vu, ce problème se pose pour la combinaison mortelle suivante.

• Système d'exploitation Windows
• TensorFlow 2.3.x

J'ai du mal à résoudre ce problème depuis un certain temps et je ne l'ai découvert que récemment.

Solution la plus simple

La solution la plus simple serait d'opter pour TensorFlow 2.4.0rcx. Il est toujours en pré-version. Mais ne m'avez pas donné le problème susmentionné. Alors,

pip uninstall tensorflow
pip install tensorflow==2.4.0rc0

Solution si vous souhaitez rester avec TF 2.3.x

Si vous êtes vraiment catégorique sur l'utilisation de TensorFlow 2.3.x, les seules solutions semblent être,

• Essayer un système d'exploitation basé sur Linux (Ubuntu)
• Compilation de TensorFlow à partir de la source (d'après ce commentaire utile )