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):
-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