Well, I don’t have any experience in using the lightning version of PyTorch but seems like a problem is occurring while transferring your data to your GPU.  Is your network architecture too big? if so try a smaller one and see if the same problem occurs or not.