Seguir as instruções aqui está funcionando para mim:
Problemas gráficos depois / durante a instalação do Ubuntu 16.04 / 16.10 com gráficos NVIDIA
Eu precisava instalar o nvidia-364 do ppa do ubuntu-graphics, eu também tinha reinstalado o Xorg antes, então isso também pode ser necessário após uma atualização.