Eu comecei a trabalhar depois de algumas tentativas (nvidia-364, do ubuntu graphics ppa).
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-364
sudo apt-get install bumblebee bumblebee-nvidia primus linux-headers-generic
sudo systemctl enable bumblebeed
Depois disso sudo gedit /etc/modules
e adicione 2 linhas:
i915
bbswitch
então
sudo prime-select intel
sudo gedit /etc/bumblebee/bumblebee.conf
na linha 22: Driver=nvidia
Certifique-se de que as linhas mudam a cada nvidia-current
para nvidia-364
(ou qualquer versão que você tenha).
Depois disso, você pode reiniciar seu computador com segurança.