The same is happening with me. Lenovo Legion y-540, Ubuntu 18, RTX-2060(6gb). My laptop automatically shuts down after a few epochs of training on GPU. I was training a multi-class text classification model using RNN+CNN. I think it’s a memory issue. But in my case, I have set the GPU only for CUDA. My laptop graphic runs on Intel Integrated HD card.
I don’t really know why this happens. And it only happens for multi-class text classification and not binary text classification (possibly a memory issue).
Any leads on this?