DNN library is not found

[MittalNeha]

When running the code for training using pytorch, I am getting this error. I am refering to the code in the Kaggle notebook
Node: 'residual_unet/conv2d/Conv2D'
DNN library is not found.
[[{{node residual_unet/conv2d/Conv2D}}]] [Op:__inference_predict_function_9480]

Any hints on what direction to take to proceed further?

python version 3.8
tensorflow: 2.8.0
cuDNN: 8.1.1
cuda: 11.2

[MittalNeha]

Resolved:
Tensorflow was pointing to a different version of cudnn. The following article helped me resolve the issue
https://macjim.medium.com/loading-alternative-cudnn-library-versions-in-tensorflow-90c7472e361a