Может ли модель, обученная на графическом процессоре, использоваться на процессоре для вывода?

Question

Я хочу сделать вывод, что моя модель использует ЦП, хотя моя машина имеет графический процессор. Хотите знать, если возможно, я могу заставить модель использовать ЦП, а не ГП? config в Tensorflow, который может принудительно использовать ЦП для вывода?

Код вывода, который я использовал:

tf.contrib.predictor.from_saved_model("PATH")

Answer 1

Предполагая, что вы используете TensorFlow 2.0, ознакомьтесь с этой проблемой на GitHub:

• [TF 2.0] Как глобально задействовать ЦП?

Решение, похоже, состоит в том, чтобы скрыть устройства GPU от TensorFlow. Вы можете сделать это, используя одну из методологий, описанных ниже:

TensorFlow 2.0:

my_devices = tf.config.experimental.list_physical_devices(device_type='CPU')
tf.config.experimental.set_visible_devices(devices= my_devices, device_type='CPU')

TensorFlow 2.1:

tf.config.set_visible_devices([], 'GPU')

(Благодарим @ymodak и @henrysky, ответивших на вопрос о проблеме с GitHub.)