StyleGAN2 output black image only with RTX 3090

I am using CUDA 11.1 with Nvidia 455 driver and Tensorflow 1.14 for the StyleGAN2 model. The resulting images are just a black screen in RTX 3090 which is unlike the case in RTX 2080ti with CUDA 10.1 and Tensorflow 1.14 where I am getting a very good result.

Anyone can please help me with this?