2016-07-01 9 views
2

die cuDNN Installationsanleitung sagtWie aktualisiere ich cuDNN auf eine neuere Version?

all platforms

Extrahieren Sie das cuDNN Archiv in ein Verzeichnis Ihrer Wahl, wie nachstehend beschrieben. Folgen Sie dann den plattformspezifischen Anweisungen wie folgt.

LINUX

cd export LD_LIBRARY_PATH = `pwd`: $ LD_LIBRARY_PATH

In den Build und Link Prozess durch -I Ihre Kompilierung Linie und -L -lcudnn zu Ihrem Link Zeile hinzufügen.

Es scheint, dass es einfach pwd-LD_LIBRARY_PATH fügt hinzu, so denke ich ersetzen nur die Dateien in pwd wird das Update tun.

Aber es scheint nicht so einfach, wie, nachdem ich dies getan habe, habe ich eine Beschwerde von Theano sagen

cuDNN Version zu alt ist. Update v5, war 3007.

Antwort

6
  1. ersetzen cudnn.h in dir/cuda/include/
  2. entfernen Sie die alte Bibliothek Dateien in dir/cuda/lib64/
  3. fügen neue Bibliotheksdateien dir/cuda/lib64/
+1

Stellen Sie außerdem sicher, dass Ihre Systempfade korrekt festgelegt sind. –

2

ich ein Drehbuch geschrieben, wo verwendet werden, um eine Version von cuDNN zu installieren oder den vorhandenen cuDNN zu einer älteren/neueren Version zu ändern. Sie können aus dem Skript herunterladen:

https://github.com/dnzzcn/cuDNNv

Dies ist, was das Skript tut:

#!/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 

Es führt die Installationsvorgänge automatisch für die Version, die Sie benötigen.

+0

unter der Annahme, 'Pakete /' ist in Ihrem 'CWD'? erfordert auch 'sudo'? – Shai

+1

Ja, der Paketordner ist im Repository-Verzeichnis enthalten. Es gibt ein Python-Skript, das das obige Bash-Skript für die gewünschte cuDNN-Version aufruft. und obwohl ich mir nicht sicher bin, sollte es ohne sudo funktionieren. – dnzzcn

Verwandte Themen