When I try using keras to train a neural network it gives me this error:
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}}]]
[[loss_2/mul/_2453]]
(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.
I updated the cudnn from 7 to 7.6 but I still get the same error, I’m trying to solve this problem but how do I check if my installation of cudnn is working?
I have a jetson nano with cuda 10 and ubuntu 18