Comenzando con azul

Serie N de Azure (GPU): instale CUDA, cudnn, Tensorflow en UBUNTU 16.04 LTS

Después de pasar más de 5 horas, encontré esta solución fácil:

-Para verificar que el sistema tenga una GPU compatible con CUDA, ejecute el siguiente comando:

lspci | grep -i NVIDIA

Verá un resultado similar al siguiente ejemplo (que muestra una tarjeta NVIDIA Tesla K80/M60):

ingrese la descripción de la imagen aquí

-Deshabilitar el controlador nouveau:

sudo -i
rmmod nouveau

-Después de un reinicio: reinicio sudo, verifique que el controlador esté instalado correctamente emitiendo:

lsmod | grep -i nvidia

-A continuación, descargue el paquete CUDA de Nvidia, …

wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb

-… hazlo saber para apt-get e instalar CUDA Toolkit:

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
sudo apt-get update
sudo apt-get install -y cuda

-Ahora podemos verificar el estado de las GPU ejecutando:

nvidia-smi

A continuación, descargamos cuDNN

wget http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz

-… descomprimir, copiar la lib64 e incluir carpetas:

tar -zxf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/
sudo rm -R cuda

-Es hora de limpiar y eliminar los archivos descargados:

rm cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
rm cudnn-8.0-linux-x64-v5.1.tgz

Para instalar Tensorflow con CPU/GPU, vaya aquí:

https://www.tensorflow.org/install/install_linux#installing_with_anaconda

Referencia:

1.https://www.lutzroeder.com/blog/2016-12-27-tensorflow-azure 2.https://www.tensorflow.org/install/install_linux#installing_with_anaconda