2016-04-18 6 views
1

Ich benutze Python mit der IDE Pycharm auf einem Ubuntu-System mit CUDA 7.5 installiert und funktioniert.CudNN Fehler beim Ausführen von TensorFlow: Konnte nicht cudnn Filter descriptor: CUDNN_STATUS_BAD_PARAM

Ich habe gerade die CUDNN-Dateien mit meiner allgemeinen CUDA-Installation zusammengeführt.

Nun, wenn ich die tiefe Faltungs nn Tutorial von TensorFlow Webseite laufen lasse, erhalte ich folgende Fehlermeldung:

F tensorflow/stream_executor/cuda/cuda_dnn.cc:383] could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM

Wer weiß, was die Ursache dieses Problems ist? Ist das ein Problem mit meiner CUDNN-Installation? (Was war Sheisty - ich lief gerade rsync -av ./include/ /usr/local/cuda/include und rsync -av ./lib64/ /usr/local/cuda/lib64, dann lief ich wieder ldconfig /usr/local/cuda/lib64) ... oder ein Tippfehler im Code?


Oder könnte es von woanders sein? (oder, noch besser, kennt jemand das Problem?)

Antwort

4

Ok, ich lief auf Null Antworten im Internet, also werde ich das lassen, falls jemand anderes auf dasselbe Problem stößt.


Das Problem war, dass cudnn v5 nicht mit Tensor Fluss funktionierte (vielleicht im Allgemeinen, vielleicht auch nur auf meinem einrichten ... die zugegebenermaßen ist Ubuntu 15.10 statt der von CUDA unterstützt 15.04), so dass ich musste zurück zu cuddn v4.

Angenommen, Sie haben die CUDA 7.5 im Standardverzeichnis installiert und den symbolischen Link in/usr/local/cuda eingerichtet. Mit CUDNN v4 (ab Feb) funktioniert das nun. Ich extrahiert nur die CUDNN tar-Datei und lief die folgenden Befehle mit Root-Rechten (aus meinem Download-Ordner):

tar -xvzf cudnn-etc.tgz 
cd cuda 

rsync -av ./lib64/ /usr/local/cuda-7.5/lib64/ 
rsync -av ./include/ /usr/local/cuda-7.5/lib64/ 

ldconfig /usr/local/cuda/lib64 
+0

Ich stieß auf das gleiche Problem am 16.04 LTS. Bitte beachten Sie, dass Sie, wenn Sie TF von der Quelle kompilieren, jetzt cudnn v5 verwenden können. Allerdings ist die pip Version von TF immer noch (Stand Juli '16) bei cuddn v4 festgefahren. Es scheint, dass Leute mit Pascal GPUs von der Quelle installieren sollten, da Pascal nur cudn v5 oder höher unterstützt. – mastazi

1

Sie verwenden die Version 5 von cudnn

Siehe Github issue #1787

Der Patch #1794 ist in Master zusammengeführt, so dass die neueste RC von Tensorflow (r0.8) has support for it.

+0

Ok, großartig, danke. Ich werde aktualisieren, sobald ich das Projekt abgeschlossen habe (alles funktioniert jetzt, werde es nicht berühren :) – bordeo

Verwandte Themen