You were able to isolate the crash to the ‘Normalize’ piece or are there other transforms as well? Can you share piece of code around this normalize part?
I have worked on Jupyter notebooks in a system with given GPU and CPU but not seen crashes…