CUDNN_STATUS_BAD_PARAM in external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc(4175): 'op' CUDNN_BACKEND_OPERATION: cudnnFinalize Failed
Hi @sylviahamidah ,

Could you please try with model.fit() instead of model.fit_generator() as it is deprecated. Also request you to share minimal code snippet if above change won't work for you. Thanks. Hi @SuryanarayanaY 
I've tried model.fit() but the error is still the same.

Here I attached the minimal reproducible example, let me know if you need further infomation [https://colab.research.google.com/drive/1FnwvnzkmeLYoaW42wkCKWw0j5xVPQ1Id?usp=sharing](url) Hi @sylviahamidah ,

Could you please confirm the cuDNN version in your system. Please cross check the compatible versions from [here](https://www.tensorflow.org/install/source#gpu).

Also, please use tensorflow[and-cuda] package which will install compatible cuda/cudnn libraries. Sorry for my late response, Sir @SuryanarayanaY 

My system version is written below. 
Tensorflow: 2.15.0
Python: 3.11.7
CUDA version: 12.2
CUDNN version: 8

Also I've tried to install tensorflow[and-cuda] with code below but the error is still the same.
`!pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings==8.6.1 tensorrt-libs==8.6.1`
`!pip install -U tensorflow[and-cuda]==2.15.0` 
  Update: After I installed `pip install tensorflow[and-cuda]`, the error disappeared. Thank you for your assistance Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/66598">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/66598">No</a>

### Issue type

Bug

### Have you reproduced the bug with TensorFlow Nightly?

No

### Source

source

### TensorFlow version

tf 2.15.0

### Custom code

Yes

### OS platform and distribution

Google Colaboratory

### Mobile device

Google Colaboratory

### Python version

3.10.12

### Bazel version

_No response_

### GCC/compiler version

_No response_

### CUDA/cuDNN version

12.2

### GPU model and memory

_No response_

### Current behavior?

When I try "model.fit()" server stops with error message below
CUDNN_STATUS_BAD_PARAM
in external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc(4175): 'op' CUDNN_BACKEND_OPERATION: cudnnFinalize Failed
	 [[{{node AlbuNet/conv1/Conv2D}}]] [Op:__inference_train_function_49877]

How can I solve the issue?
Thanks!

### Standalone code to reproduce the issue

```shell
you can see my work here
https://colab.research.google.com/drive/1n97wV5cfgdQfWmngK5EBJjsYSjJzfi_d?usp=sharing
```


### Relevant log output

_No response_