web-dev-qa-db-fra.com

ValueError «Vous essayez d'utiliser l'ancien back-end GPU» lors de l'importation de keras

J'utilise Keras avec le backend Theano sur Ubuntu 16.04. Ma configuration a fonctionné sans problème, cependant, tout d'un coup, j'obtiens l'erreur suivante lorsque j'importe Keras (import keras):

ValueError: vous essayez d'utiliser l'ancien backend GPU. Il a été retiré de Theano. Utilisez device = cuda * maintenant. Voir https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29 pour plus d'informations.

Comment résoudre ça?

12
Tshilidzi Mudau

Vous devez modifier (ou ajouter) votre variable d'environnement appelée THEANO_FLAGS. Si vous définissez la variable de sorte qu'elle contienne device=cuda au lieu de device=gpu l'erreur disparaîtra.

Plus d'informations sur cette variable peuvent être trouvées ici .

17
Wilmar van Ommeren