Quería probar un poco CUDA en mi notebook Lenovo, así que manos a la obra.

Obs:

•  Utilizo Ubuntu 16.04 LTS Xenial Xerus de 64 bits.
•  Asumo que se tiene el driver oficial de NVIDIA instalado.

Entramos a la terminal para verificar la tarjeta NVIDIA:
elspci | grep -i nvidia

Sale por pantalla:
3c:00.0 3D controller: NVIDIA Corporation Device 1d10 (rev a1)

Volvemos a teclear en la terminal:
sudo apt update
sudo apt install linux-headers-$(uname -r)

Entramos a este sitio y descargamos cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64.deb
Volvemos a teclear en la terminal:
sudo dpkg -i cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64.deb
wget -qO - http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/7fa2af80.pub | sudo apt-key add -
sudo apt update
sudo apt install cuda
sudo apt clean && sudo apt autoclean

Configurar CUDA.
Volvemos a teclean en la terminal:
cd /etc/profile.d/
sudo nano cuda.sh

Agregamos lo siguiente:
export PATH=$PATH:/usr/local/cuda/bin
export CUDADIR=/usr/local/cuda
export GLPATH=/usr/lib

Guardamos los cambios (control + o) y luego cerramos el editor. Volvemos a teclean en la terminal:
cd /etc/ld.so.conf.d/
sudo nano cuda.conf

Agregamos lo siguiente:
/usr/local/cuda/lib64

Guardamos los cambios (control + o) y luego cerramos el editor. Volvemos a teclean en la terminal:
sudo ldconfig
sudo reboot

Ahora podemos usar CUDA en nuestro equipo.
Verificar la versión de CUDA:
nvcc -V

Sale por pantalla:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

Imagen destacada: Genbeta dev
Fuente 1: Mathalope
Fuente 2: lithiumdenis