Изменение количества графических процессоров при развертывании

Question

У меня есть настроенная начальная версия 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 графический процессор?

Answer 1

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

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