'Loaded runtime CuDNN library: 8.0.5 but source was compiled with: 8.1.0 , Tensorflow object-detection API

I am trying to train an Object detector using Tensorflow object-detection API on Colab, This is not the first time i am doing it and have trained multiple models using the same API. Today suddenly I am getting the following error "Loaded runtime CuDNN library: 8.0.5 but source was compiled with: 8.1.0.", I dont know what to do about this.

I tried downgrading TF version to 2.4,but that did'nt work too. I dont know how we can downgrade/upgrade the CudaNN in Google colab.

As a result of the above error I am also getting the following error "tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize"

Here's the link to my colab notebook - link

Any help would be appreciated.



Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source