[GPU] I need to train 1024x1024 model What GPU should I have?

[Alde]

Let's say money is no issue. Wil be fine to buy RTX A6000 48 gb or I need go with A100 80GB?
Thank you for your answers.

[torzdf]

That's a hard question to answer, and given the amount of $$$ you are about to drop, I'd be loathed to steer you in any particular direction.

However, it is possible to train 1080px on an 11GB card using the DNY1024px model. However, I cannot speak for the quality of the final output, because I do not know of anyone who has had the patience to train at that resolution to completion.

Higher quality models are going to need significantly more VRAM (obviously). If money was no issue, I would go with the largest I could to make sure I was as future proofed as possible. However, some testing, trial and error is going to be required on your part to create a model that gives you the output you require.

[Ryzen1988]

Let's say money is no issue. Wil be fine to buy RTX A6000 48 gb or I need go with A100 80GB?
Thank you for your answers.

I have a a6000, 48GB is enough for a 1024px but training time is horrible, so i would imagen a100 or h100 is way faster