Install the last version of nvidia & cuda for ubuntu

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-381
sudo apt-get install nvidia-cuda-toolkit
ldconfig -p | grep cuda