How much graphics memory will be spent?

[sea-with-stars]

I train the net with UIEBD datasets but I cannot work it out with A100, which has 40GB memory.