Как проверить, является ли кеш-сервер тензорным потоком графическим процессором или версией процессора?

Я понимаю, что при установке tensorflow вы либо устанавливаете версию графического процессора, либо версию CPU. Как проверить, какой из них установлен (я использую linux).

Если версия GPU установлена, будет ли она автоматически запускаться на CPU, если графический процессор недоступен или выйдет ошибка? И если доступен графический процессор, есть ли определенное поле или значение, которое необходимо настроить, чтобы убедиться, что он работает на GPU?

Ответ 1

Также вы можете проверить с помощью функции Keras backend:

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

Я проверяю это на Keras (2.1.1)

Ответ 2

Согласно документации.

Если вы используете серверы TensorFlow или CNTK, ваш код будет автоматически запускаться на GPU, если обнаружен какой-либо доступный графический процессор.

Вы можете проверить, что все устройства используются тензорным потоком,

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

Также, как предложено в этом ответе

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

Это будет печатать, использует ли ваш тензор поток процессора или GPU. Если вы используете эту команду в ноутбуке jupyter, проверьте консоль, откуда вы запустили ноутбук.

Если вы скептически относитесь к тому, что вы установили версию tenorflow gpu или нет. Вы можете установить версию gpu через pip.

pip install tensorflow-gpu