При запуске алгоритма распознавания лиц с использованием Python API TensorRT (а также PyCUDA) я сталкиваюсь с ошибкой ниже:
[TensorRT] ERROR: ../rtSafe/safeContext.cpp (133) - Cudnn Error in configure: 7 (CUDNN_STATUS_MAPPING_ERROR)
[TensorRT] ERROR: FAILED_EXECUTION: std::exception
Код по-прежнему компилируется и выполняется, но результаты неточные -выходной сигнал программы колеблется от 0,9999990907141816 до 0, когда ожидается более непрерывный диапазон чисел. Я протестировал это с TF-TRT и Keras, и мой код работает в обоих (с небольшими изменениями, чтобы соответствовать различиям между API TF и Keras).
Я пытался установить разные версии CUDA (9.0,10.0 и 10.1) и CuDNN (7.6.3, 7.6.5). Версия TensorRT - 6.0.1.5, а PyCUDA - 2019.1.2. Если это поможет, я запускаю это в Ubuntu 18.04.
Любая помощь будет принята!
Обновление: Я думаю, что ошибка вызвана одновременным запуском сеанса TensorFlow. В частности, я использую пакет mtcnn ( link ), который может мешать работе TensorRT. Когда mtcnn инициализирует сеанс TF, возникает вышеуказанная ошибка;когда mtcnn не используется, эта ошибка не возникает и все работает, как ожидалось.