Como atualizar o Tensorflow para v1.3 (atualização cuDNN & CUDA)

Question

Primeiro, abaixo está o meu ambiente de sistema

1. SO: Ubuntu 16.04
2. current tensorflow: 1.0.1
3. cuDNN atual: 5.1.5
4. CUDA atual: 8,0

Eu quero atualizar tensorflow para 1.3, mas tf 1.3 requer cuDNN 6.

Portanto, é necessário atualizar o cuDNN, mas não consigo encontrar o modo de atualizar cuDNN e configurar o tensorflow como eu instalei antes.

    

Answer 1

Eu escrevi um script que pode ser usado para limpar a instalação de uma versão desejada do cuDNN ou atualizar uma existente. Depois de executar as etapas na página do GitHub, você pode alterar sua versão do cuDNN para a v6.0. Depois disso, você não precisa fazer nenhuma configuração para o TensorFlow, ele começará a funcionar automaticamente com o cuDNN v6.0. Você pode encontrar o script aqui .

Isso é o que o script faz:

#!/bin/bash

rm -f /usr/include/cudnn.h
rm -f /usr/lib/x86_64-linux-gnu/*libcudnn*
rm -f /usr/local/cuda-*/lib64/*libcudnn*


cp -P packages/cudnn/include/cudnn.h /usr/include
cp -P packages/cudnn/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*

rm -rf packages/cudnn

Executa operações de instalação automaticamente para a versão desejada.