Why do you need a drift, are you trying to change the bearings at the same time?
1) remove cassette
2) remove end caps (I use either black & decker work mate, moll grips gently, or vice gently – i.e. firm enough to grip, but not firm enough to mark the ends)
3) remove freehub body (twist & pull)
4) put hub over suitable support – I use my b&d workmate
5) using a rubberised mallet, tap axle out from freehub side – the rotor side bearing will also drift out, as will the green rubber seal
6) put long end through workmate hole leaving bearing well supported and stubby end of axle showing
7) gentle tap axle through so bearing is free
to re-fit is reverse of above.
If you really need to drift the bearing out, or for refitting the rotor side bearing to the new axle, most common socket sets will have a suitable sized socket. If you’re using a rubber mallet its usually ok. However, I usually just turn it over and use workmate as a support for the bearings. Obviously all at your own risk.