Ошибка после прогнозирования с использованием модели CNN, загруженной с диска

Question

Я загрузил свою модель CNN, используя

model = load_model('./model/model.h5')

Эта модель загружается хорошо (хотя я получаю предупреждающие сообщения: «Вызов экземпляра инициализатора с аргументом dtype вместо передачи его в constructor ")

Однако, когда я пытаюсь предсказать, используя эту модель, я получаю следующую ошибку:

UnknownError: 2 root error(s) found.
  (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d/Conv2D}}]]
     [[dense_2/Softmax/_273]]
  (1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d/Conv2D}}]]
0 successful operations.
0 derived errors ignored

Есть идеи, как решить эту проблему?

Answer 1

Видимо, в моей системе просто не было установлено CUDA. Сделал это, и теперь он работает!