You do need to get the paperwork sorted out in VIE as that's where you are leaving the EU.
You will go through customs in ZRH, though that will be just spot checks and you may well not even see anyone. There are no formal passport checks but the customs may well do one if they decide to do a spot check.