Eu consegui funcionar depois de algumas tentativas (nvidia-364, da 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 alterar as linhas, mude nvidia-current
paranvidia-364
(ou qualquer versão que você tenha).
Depois disso, você pode reiniciar o computador com segurança.