tensorflow:未定义的符号:cudnnSetRNNDescriptor_v6

时间:2017-11-13 02:05:38

标签: python-3.x tensorflow

当我完成使用virtualenv环境安装tensorflow(GPU_support,linux 14.04,python3.4)时,在官方网站的指示下,我使用以下命令验证安装:python;导入张量流;但是有一个错误:

  • 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 ***

1 个答案:

答案 0 :(得分:3)

我有同样的错误,所以希望这个解决方案对你有用......

您使用的是什么版本的CuDNN?我发现tensorflow-gpu&gt;的版本当我安装了CuDNN v5.1.10时,1.2无法加载。

当我写这篇文章时,Tensorflow文档说你必须拥有CuDNN v6。 https://www.tensorflow.org/install/install_linux#nvidia_requirements_to_run_tensorflow_with_gpu_support

所以我刚刚删除了旧的CuDNN二进制文件和标题:

$ 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

以及v5.1.10的任何其他残余。然后只需将新的v6标题和二进制文件下载并复制到相同的位置。

这是检查CuDNN版本的有用命令:

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

相关问题