Suggession regarding a GPU to use for multi class sementic segmentation!

Hey guys!

I hope you all are doing well. I am currently stuck in a position where I have to make a decision on which GPU I must buy for a multiclass image segmentation problem. I never had to choose between a wide variety of GPUs until today for training a network. However, there is a GPU called Nvidia RTX A4000 16GB which perfectly fits my budget, but I am now unsure if this would handle the workload. Therefore, I would like to get a piece of advice from professionals or from people who might have used this GPU, if it is good for this problem statement?

CNN Networks that I will be exploring are UNet, ResNet and AlexNet.

Any suggestion/recommendation would be highly appreciated.

Thank you in advance and best regards,