Para instalar los drivers de video en Ubuntu tienes que ejecutar los siguientes comandos.
ubuntu-drivers devices
que nos brinda la informacion de la tarjeta de video disponible, buscamos los drivers correspondientes con el comando.
sudo ubuntu-drivers install
Instalar el tookit de cuda
sudo apt install nvidia-cuda-toolkit
Hay que hacer reboot y activar los drivers.
sudo reboot
Con ello, ya tendriamos los drivers de nvidia listo en Ubuntu
puedes verificar ello ejecutando.
nvidia-smi

Saludos