все.У меня проблема, когда я использую Keras с бэкендом cntk-gpu версии 2.5.1.
По какой-то причине я должен использовать cntk-gpu 2.5.1 в качестве бэкэнда Keras, и у меня есть фрагмент кода сОсновной код выглядит следующим образом (действительно простой код для прогнозирования):
# test_x: test data from dataset imagenet
predict_model = keras.models.load_model(model_path,custom_objects=ModelUtils.custom_objects())
predict_model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
res = predict_model.predict(test_x,batch_size=4)
Я обнаружил, что версия cntk-gpu2.7 занимает всего 50 секунд, а версия 2.5.1 - 20 минут.Консоль показывает, что cntk использует gpu, потому что она печатает информацию о моем GPU, например:
Выбранный графический процессор [0] GeForce GTX 1080 Ti в качестве устройства по умолчанию для всего процесса
Iпротестировали как на ubuntu18.04 cuda10.0, так и на ubuntu16.04 cuda9.1.Версия 2.4 и ниже также столкнутся с этой проблемой
Я не знаю, в чем причина и как решить проблему.Вы когда-нибудь сталкивались с ситуацией, когда версия cntk-gpu 2.5 работает очень медленно?
Я надеюсь услышать от вас.Заранее спасибо.