I am using Ubuntu 18.04. I got NVIDIA GTX 1060 with nvidia drivers v415. Every time I train the model on GPU while on battery power, system automatically shuts down. But, when I am connected to power, it doesnt happen. I am not sure as to what is causing this problem. Has anyone faced such a problem ?
Any help would be appreciated.