Para instalar os drivers da Nvidia, basta abrir
Settings -> Software & Updates -> Additional Drivers
dê alguns segundos para atualizar automaticamente sua lista e escolha a versão mais recente do driver da Nvidia ... atualmente 367.35, que é estável, mas muito recente
Após a reinicialização, você pode instalar o SDK CUDA
PS. Se você ignorar o conselho acima e, em vez disso, instalar os drivers da Nvidia manualmente (o que eu fiz muitas vezes), no início, todos receberão a mensagem de erro: "script de pré-instalação falhou" ... ignore esse aviso, pois NÃO tem nenhum sinal de sucesso ou falha da instalação