Instalação CUDA 9.0/cuDNN 7.1 para Ubuntu 16

Olá pessoal!

Compartilho com vocês minha experiência instalando o CUDA Toolkit 9.0 e o cuDNN 7.1 para Linux Ubuntu 16

As instruções oficiais estão aqui: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

Um resumo (que eu fiz) segue:

1. Instalação do Driver Nvidia.

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update
sudo apt install nvidia-387

Confira se o Drive foi instalado com:

$ lsmod | grep nvidia

2. Instalação do CUDA Toolkit

Baixe e execute o arquivo do CUDA Toolkit baixado do site da Nvidia:

Run `sudo sh cuda_9.0.176_384.81_linux.run`

Inclua nas variáveis de ambiente PATH e LD_LIBRARU_PATH .

Para fazer isso permanentemente : `gedit ~/.profile’, no seu diretório Home e inclua as linhas abaixo no final do arquivo.

# CUDA
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

Como o driver foi instalado, quando solicitado diga (n)ão para a instação do driver.

Instale cuDNN conforme: http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html

Teste CuDNN realizado com sucesso como MNIST!