У меня есть сборка Mobilenet на Keras.Локальный запуск занимает около 290 секунд каждый шаг, но когда я запускаю на GCLoud ML Engine, это занимает более 400 секунд. Я добавляю следующую строку в мой код:
K.tensorflow_backend._get_available_gpus()
И журнал выглядит следующим образом:
['/job:localhost/replica:0/task:0/device:GPU:0']
Я также пытался перейти с 1 GPU на 4 GPU, но результат тот же.Нужно ли менять какой-либо код для оптимизации Keras на GPU?