【发布时间】:2018-10-25 22:42:29
【问题描述】:
我刚刚安装了
Ubuntu 18.04
NVIDIA GPU 驱动程序 390
CUDA Toolkit 9.0 版(包含所有补丁)
CUDNN 7.0.5
而不是 conda 用于 tensorflow 安装 我用过 Virtualenv
安装后,我无法在 python 中导入 tensorflow 模块 由于这个错误
Python 2.7.15rc1 (default, Apr 15 2018, 21:51:34)
[GCC 7.3.0] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/kai/tensorflow/local/lib/python2.7/site-packages/tensorflow/__init__.py", line 24, in <module>
from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
File "/home/kai/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
File "/home/kai/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 74, in <module>
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "/home/kai/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "/home/kai/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "/home/kai/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory
但我认为我正确地将它们放在正确的位置,如下所示
$ cd /usr/local/cuda/lib64
:/usr/local/cuda/lib64$ ls -l libcudnn*
-rwxr-xr-x 1 root root 287624224 May 15 19:01 libcudnn.so
-rwxr-xr-x 1 root root 287624224 May 15 19:01 libcudnn.so.7
-rwxr-xr-x 1 root root 287624224 May 15 19:01 libcudnn.so.7.0.5
-rw-r--r-- 1 root root 280805558 May 15 19:02 libcudnn_static.a
所以现在我想知道为什么会发生这种情况
感谢阅读本文
【问题讨论】:
-
你有没有导出 LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64 export CUDA_HOME=/usr/local/cuda
-
如果这仍然是一个问题 - 建议在 stackoverflow.com/a/61563579/1215913 检查我的答案,看起来你和我有同样的问题 - 符号链接已损坏..
标签: python python-2.7 tensorflow