Hi @Mike Newton
You can use the virtual machines selector to find the best for your purpose:
https://azure.microsoft.com/en-us/pricing/vm-selector/
And you can lear more about GPU Family on:
https://learn.microsoft.com/en-us/azure/virtual-machines/sizes-gpu
Get in touch if you need more help with this issue.
--please don't forget to "[Accept the answer]" if the reply is helpful--