web-dev-qa-db-fra.com

Comment vérifier si keras tensorflow backend est une version GPU ou CPU?

Je comprends que lors de l’installation de tensorflow, vous devez installer la version GPU ou CPU. Comment puis-je vérifier lequel est installé (j'utilise linux).

Si la version du GPU est installée, s’exécutera-t-il automatiquement sur le processeur si le GPU n’est pas disponible ou génère-t-il une erreur? Et si le GPU est disponible, y a-t-il un champ ou une valeur spécifique que vous devez définir pour vous assurer qu'il est exécuté sur le GPU?

tensorflowkeras
40
matchifang

Aussi, vous pouvez vérifier en utilisant la fonction backend de Keras:

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

Je teste cela sur Keras (2.1.1)

55
Bumseok

Selon le documentation .

Si vous exécutez le backend TensorFlow ou CNTK, votre code sera automatiquement exécuté sur le GPU si un GPU disponible est détecté.

Vous pouvez vérifier ce que tous les appareils sont utilisés par tensorflow en -

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

Aussi comme suggéré dans cette réponse

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

Cela vous indiquera si votre tensorflow utilise un processeur ou un backend GPU. Si vous exécutez cette commande dans jupyter notebook, consultez la console à partir de laquelle vous avez lancé le notebook.

Si vous êtes sceptique si vous avez installé la version gpu de tensorflow ou non. Vous pouvez installer la version gpu via pip.

pip install tensorflow-gpu

19
markroxor