Я загрузил свою модель 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
Есть идеи, как решить эту проблему?