Memory Issue

Dear NVIDIA Team,

I have RTX 3090. I am trying to train a deep learning model on images of size 512x512x3 but it is giving out of memory error on batch size other than 1.

Please suggest any solution.
Let me know if any other information is required.