Error with Tutorial in Google Colab and GPU: "Failed to get convolution algorithm"

Viewed 288

when I run this tutorial https://www.tensorflow.org/lite/tutorials/model_maker_object_detection on a GPU I get the following error: "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."

  1. I do not have any warning log messages printed above the error message.
  2. No issues if I run on a CPU.
  3. I do not make any changes to the tutorial before execution.
  4. Searching online I found that the problem may disappear if I reset the runtime. It does not work for me.
  5. Searching online I found that the problem may be with the version of cuDNN. Before spending time to change the cuDNN version, I would like to know your opinions.

Please let me know if you need further information.

Thanks, Federico

0 Answers
Related