Keras 2.2.4 с TensorFlow 1.4.1 сбой экземпляров графического процессора

Question

Какое минимальное требование к версии TensorFlow для Keras версии 2.2.4?

У меня проблемы при использовании архитектуры Conv2D, кажется, что экземпляр GPU падает, то есть я вижу, как память GPU заполняется немного, а затем запущенные процессы просто «сбой».ошибка, ноутбук просто «зависает».Тренировочные плотные модели например отлично работают.Этот точно такой же ноутбук с архитектурой Conv2D прекрасно работает на моем ноутбуке с TensorFlow 1.12.0 & Keras 2.2.4.

Я ожидаю, что это как-то связано с используемой версией Keras & TensorFlow.Используется графический процессор Tesla M10 (который поддерживает только CUDA 8.0?).Сервер с этим M10 имеет Tensorflow версии 1.4.1 и Keras 2.2.4.

Буду признателен за любые идеи по решению этой проблемы.

Answer 1

Совместимость версий между keras и tensorflow - это проблема, с которой, вероятно, сталкивался кто-либо.

Как и в моем ответе здесь , вы можете использовать одну комбинацию tensorflow-gpu 1.4 иkeras 2.0.8.Вы также можете проверить здесь , чтобы узнать больше комбинаций.

Если вам нужно использовать keras 2.2.4, вам придется установить tensorflow-gpu 1.11 и более поздние версии, для чего требуется cuda 9.