DNN library is not found

Open MittalNeha opened this issue 2 years ago • 1 comments

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

Jun 13 '23 14:06 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

Jun 14 '23 06:06 MittalNeha