Tensorflow crash driver CUDA of GeForce RTX 4090

I have two Geforce RTX 4090 on one motherboard. I’m running two processes for training Tensorflow models on different maps. After some time, the process crashes on GPU 0, but on GPU 1 it works without failures. I am attaching a screenshot of the error. How to fix it?
Driver Version : 537.34
CUDA Version : 12.2
TensorFlow version: 2.10.0
OS: Win10x64 Pro

Hi @user267, As per test build configuration Tensorflow 2.10 supports CUDA 11.2. But I can see that you are using CUDA 12.2. Could please try with the cuda 11.2. Thank You.