how - ¿Cómo verificar si el backend keras tensorflow es la versión de GPU o CPU?

keras check if using gpu

Question

Esta pregunta ya tiene una respuesta aquí:

• ¿Cómo compruebo si keras está utilizando la versión gpu de tensorflow? 3 respuestas

Entiendo que al instalar tensorflow, instala la versión de GPU o CPU. ¿Cómo puedo comprobar cuál está instalado (uso linux).

Si la versión de GPU está instalada, ¿se ejecutaría automáticamente en la CPU si GPU no está disponible o arrojaría un error? Y si la GPU está disponible, ¿hay algún campo o valor específico que deba establecer para asegurarse de que se está ejecutando en la GPU?

Answer 1

Según la documentation .

Si se está ejecutando en los backends TensorFlow o CNTK, su código se ejecutará automáticamente en la GPU si se detecta alguna GPU disponible.

Puede verificar que tensorflow utiliza todos los dispositivos mediante -

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

También como se sugiere en esta respuesta.

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

Esto imprimirá si su tensorflow está utilizando una CPU o un backend de GPU. Si está ejecutando este comando en la notebook jupyter, revise la consola desde donde lanzó la notebook.

Si eres escéptico si has instalado la versión gpu tensorflow o no. Puede instalar la versión gpu a través de pip.

pip install tensorflow-gpu

Answer 2

También puedes verificar usando la función de backend de Keras:

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

Pruebo esto en Keras (2.1.1)