Eu estava tendo o mesmo problema e corrigi-lo executando os seguintes comandos no terminal:
sudo apt-get purge nvidia* ## remove the previously installed proprietary driver
sudo reboot
sudo add-apt-repository ppa:xorg-edgers/ppa
sudo apt-get update
sudo apt-get install nvidia-340
Isso funcionou para mim. Deixe-me saber como isso funciona para você.