You can assume your spouse's surname at any point, you just need to show to marriage certificate when you change ID and show the name change and new ID to your bank and other accounts.

It's a pain in the ass if things don't match, so it's best to do everything or nothing.