Upgrading Cudnn version in Vertex AI Notebook [Kernel Restarting Problem]-CodePudding

Problem: Cudnn version incompatiable with tensorflow and Cuda, Kernel dies and unable to start training in Vertex AI.

Current versions:

import tensorflow as tf
from tensorflow.python.platform import build_info as build
print(f"tensorflow version: {tf.__version__}")
print(f"Cuda Version: {build.build_info['cuda_version']}")
print(f"Cudnn version: {build.build_info['cudnn_version']}")

tensorflow version: 2.10.0
Cuda Version: 11.2
Cudnn version: 8

As per the information (shown in attached screenshot) available

A similar question has been asked