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
说明 cudnn版本不对。
查看cudnn版本
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
移除已有的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
然后按照http://blog.csdn.net/lucifer_zzq/article/details/76675239
所示步骤安装即可。
更改链接:
sudo ln -sf /usr/local/lib/libcudnn.so.6.0.21 /usr/local/lib/libcudnn.so.6
sudo ln -sf /usr/local/lib/libcudnn.so.6 /usr/local/lib/libcudnn.so
sudo ldconfig -v