Pessoal, obrigado pela ajuda. No final foi mais simples que pensei. As informações aqui me levou a solução.
Instalei o um ppa pelo terminal com o comando:
sudo add-apt-repository ppa:kelebek333/nvidia-legacy
Depois atualizei:
sudo apt-get update
Depois usei esse comando:
sudo apt install nvidia-340
Depois reiniciei pelo terminal mesmo:
reboot
E já voltou com drive correto.
Não sei se fez diferença, mas já instalei a distro com drives da Nvidia.
Espero que esse tópico ajude outros com essa mesma dificuldade.