Eu vi outras respostas, mas não ajudou muito. Aqui está como eu fiz isso.
Você precisa baixar todos os três arquivos do site da Nvidia Developer. link
imagem mostrando os arquivos que você precisa baixar
mas o problema é que você precisa wget desses arquivos.
Bem, você pode fazer isso daqui:
link
Use Ctrl + F para pesquisar nomes de arquivos exatos que você pode obter dos links do site da Nvidia Developer.
Esse site ftp não possui o arquivo .deb do DOC (último - libcudnn7-doc_7.1.3.16-1 + cuda9.1_amd64.deb). Você pode fazer o download aqui: link
Obrigado Universidade de Chicago:).
Depois disso, é só seguir o guia de instalação de link :
Do diretório onde você acabou de baixar os 3 arquivos contendo cuDNN instale as bibliotecas de tempo de execução, por exemplo:
sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb
Instale a biblioteca do desenvolvedor, por exemplo:
sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb
Instale os exemplos de código e o Guia do usuário da biblioteca cuDNN, por exemplo:
sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb
Use seus próprios arquivos principais, dev e doc em vez daqueles nos exemplos.