Could it be possible that u loaded other things in the CUDA device too other than the training data features, labels and the model
Deleting variables after training start won’t help coz most variables are stored and handled on the RAM and cpu except the ones specified on the CUDA enabled gpu which should be just training data and model
Pls can u post ur code if possible