alienware 18r2 problema al entrenar modelos con GPU python

Hola, estoy entrenando un modelo LTSM y si utilizo tensorflow o pytorch para entrenar el modelo se me crashea el kernel a los pocos minutos de haber comenzado, solo funciona si utilizo solo la CPU a alguien mas le paso? drivers actuales  556.12