2017-11-13 2 views
4

als ich die installation tensorflow (GPU_support, linux 14.04, python3.4) mit virtualenv umgebung, nach den anweisungen der offiziellen website, validierte ich die installation mit dem befehl: python; Import Tensorfluss; aber es ist ein Fehler:tensorflow: undefined symbol: cudnnSetRNNDescriptor_v6

  • import tensorflow as tf Traceback (most recent call last): File "/home/fangfang/tensorflow/lib/python3.4/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module> from tensorflow.python.pywrap_tensorflow_internal import * File "/home/fangfang/tensorflow/lib/python3.4/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module> _pywrap_tensorflow_internal = swig_import_helper() File "/home/fangfang/tensorflow/lib/python3.4/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description) File "/home/fangfang/tensorflow/lib/python3.4/imp.py", line 243, in load_module return load_dynamic(name, filename, file) ***ImportError: /home/fangfang/tensorflow/lib/python3.4/site-packages/tensorflow/python/../libtensorflow_framework.so: undefined symbol: cudnnSetRNNDescriptor_v6 ***

Antwort

2

ich diesen gleichen Fehler hatte, und so hoffentlich diese Lösung für Sie arbeiten ...

Welche Version von CuDNN sind Sie benutzen? Ich habe festgestellt, dass Versionen von tensorflow-gpu> 1.2 nicht geladen werden konnten, während ich CuDNN v5.1.10 installiert hatte.

Zu der Zeit, als ich dies schreibe, sagen Tensorflow-Dokumente, dass Sie CuDNN v6 haben müssen. https://www.tensorflow.org/install/install_linux#nvidia_requirements_to_run_tensorflow_with_gpu_support

So nahm ich nur die alten CuDNN Binärdateien und Header:

$ sudo rm /usr/local/cuda/include/cudnn.h

$ sudo rm /usr/local/cuda/lib64/libcudnn.so

$ sudo rm /usr/local/cuda/lib64/libcudnn.so.5.1.10

und andere Reste von v5.1.10. Dann lade einfach die neuen v6 Header und Binärdateien herunter und kopiere sie an die gleichen Stellen.

Dies ist ein nützlicher Befehl Ihre CuDNN Version zu überprüfen:

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

+0

i installiert tensorflow für python2.7 von Anaconda vor und es funktioniert gut. Ich habe überprüft, meine Cuda-Version ist 8. Dieses Mal habe ich Tensorflow für python3 installiert, ich denke, meine cuddn Version hat kein Problem. Ich weiß nicht warum – bangbangbang

+0

Und welche Version von CuDNN (nicht CUDA) hast du? Es gibt einen Befehl, der oben für das Drucken auf der Konsole bereitgestellt wird. –

+0

ja, du hast Recht. Es ist das Problem des Kuschels. und ich fand ein sehr nützliches Werkzeug, um die Version von cudnn zu ändern. [link] (https://github.com/dnzzcn/cuDNNv) – bangbangbang

Verwandte Themen