每个人。我在后端cntk-gpu版本2.5.1中使用Keras时遇到问题。
出于某种原因，我必须将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信息，例如：
选择GPU [0] GeForce GTX 1080 Ti作为进程范围的默认设备
我已经在ubuntu18.04 cuda10.0和ubuntu16.04 cuda9.1上进行了测试。 2.4及以下版本也会遇到此问题
我不知道原因是什么以及如何解决问题。 您是否遇到过cntk-gpu 2.5版本运行非常缓慢的情况？
我希望收到你的来信。预先谢谢你。