2017-02-23 5 views
1

У меня есть настроенная начальная версия v3 на 2-GPU-машине. Теперь я пытаюсь запустить обученную модель на другой машине с 1 GPU, но я получил сообщение об ошибке, как это:Изменение количества графических процессоров при развертывании

Невозможно назначить устройство к узлу «tower_1/градиенты/tower_1/CONV0/BatchNorm/моменты/enough_statistics/Sub_grad/BroadcastGradientArgs ': Не удалось удовлетворить явную спецификацию устройства/устройство: GPU: 1', потому что в этом процессе не регистрируются устройства, соответствующие этой спецификации; доступные устройства:/работа: локальная/реплика: 0/задача: 0/CPU: 0,/работа: локальная/реплика: 0/задача: 0/GPU: 0

Кажется, что модель хочет 2 -GPU, как тот, на котором он был обучен. Могу ли я преобразовать эту модель, чтобы она использовала только 1 графический процессор?

ответ

0

Я изменил две вещи, и это сработало.

  1. Включите allow_soft_placement вариант в сессии:

    config = tf.ConfigProto(allow_soft_placement=True) 
    sess = tf.Session(config=config) 
    
  2. Rename имя файла модели из model.ckpt-50000.data-00000-of-00001 в model.ckpt-50000

Смежные вопросы