AMP uses more GPU memory and slows training

I would suggest to double-check your PyTorch version on the laptop (by printing torch.__version__) to make sure that you got a very fresh relelease/nightly build. It seems really easy to have conda trick you into using an old version.

As explained by @ptrblck , here, there was a bug not too long ago:

Best regards

Thomas