Since you have upgraded the OS and here I assume that you have done all the settings check on windows powers options(access by clicking on the battery icon on your laptop) to stay powered on as always on battery and ac power.
As per for the driver, it is worth to give it a try, but still, go for it. Here I would recommend you to check if the BIOS fan( exhaust fan) on the laptop is working fine. [br]
have you tried to clean the vents for dust? As the overheated BIOS attempts the laptop to shut down to avoid major damages caused by overheating.
As you have already used a different OS version to solve the problem, it seems to be the vent issues. Check for it.