Instale os drivers da Nvidia por
sudo apt install nvidia-361
e reinicie. Se o UEFI for usado, desabilite o Secure Boot.
Para reduzir o aquecimento, você pode alternar para o adaptador integrado
sudo prime-select intel
faça logoff ou reinicie.