【发布时间】:2016-07-09 14:58:17
【问题描述】:
我在 Ubuntu 15.10 上运行 TensorFlow。当我输入pip show tensorflow时,我看到TF已经正确安装了。
但是,当我写 import tensorflow as tf 时,我收到以下错误消息:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/__init__.py", line 23, in <module>
from tensorflow.python import *
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 49, in <module>
from tensorflow import contrib
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/contrib/__init__.py", line 23, in <module>
from tensorflow.contrib import layers
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/contrib/layers/__init__.py", line 68, in <module>
from tensorflow.contrib.layers.python.layers import *
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/contrib/layers/python/layers/__init__.py", line 22, in <module>
from tensorflow.contrib.layers.python.layers.initializers import *
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/contrib/layers/python/layers/initializers.py", line 24, in <module>
from tensorflow.python.ops import random_ops
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/python/ops/random_ops.py", line 23, in <module>
from tensorflow.python.framework import ops
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 39, in <module>
from tensorflow.python.framework import versions
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/python/framework/versions.py", line 22, in <module>
from tensorflow.python import pywrap_tensorflow
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module>
_pywrap_tensorflow = swig_import_helper()
File "/home/me/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow', fp, pathname, description)
ImportError: libcudart.so.7.5: cannot open shared object file: No such file or directory
为了它的价值,我已经按照here 的说明设置了我的LD_LIBRARY_PATH 和CUDA_HOME 环境变量。
有什么建议吗?
编辑:
我已经安装了 CUDA 7.5 并将这些添加到我的 .profile 文件中:
export LD_LIBRARY_PATH="/usr/local/cuda-7.5/lib64"
export CUDA_HOME=/usr/local/cuda-7.5
但是,我仍然看到相同的错误消息。
编辑:
我在运行ldd /usr/local/cuda-7.5/lib64/libcudart.so.7.5 时看到以下输出:
linux-vdso.so.1 => (0x00007ffdac7ea000)
libc.so.6 => /lib/x86_64-linux-gnu/libc.so.6 (0x00007fc27a281000)
libdl.so.2 => /lib/x86_64-linux-gnu/libdl.so.2 (0x00007fc27a07d000)
libpthread.so.0 => /lib/x86_64-linux-gnu/libpthread.so.0 (0x00007fc279e5e000)
librt.so.1 => /lib/x86_64-linux-gnu/librt.so.1 (0x00007fc279c56000)
/lib64/ld-linux-x86-64.so.2 (0x00005604f5406000)
编辑:
如果相关,我使用 GeForce GT640。
编辑:
在运行 TensorFlow 之前,我关注了 @tommus' advice 并调用了 source ~/.profile,现在它就像一个魅力。
感谢所有在 cmets 中尝试帮助我的人——这是我第一次使用任何类型的 Linux 发行版,我真的非常感谢我收到的所有耐心帮助远 :-) 你们真是太棒了!
【问题讨论】:
-
tensorflow 在这种情况下正在寻找 CUDA 7.5。您安装的是 CUDA 7.5 还是 CUDA 7.0?您将
LD_LIBRARY_PATH设置为哪条路径?那条路径上是否有 libcudart.so.7.5? -
嗨!我安装了 CUDA 7.0,并将其添加到我的 .profile 文件中:
export LD_LIBRARY_PATH="/home/me/cuda/targets/ppc64le-linux/lib" -
尝试安装 CUDA 7.5。
-
@RobertCrovella 谢谢,我会试一试。如果我遇到任何问题,我会再次发布:-)
-
@RobertCrovella 我应该写什么命令来找出 CUDA 7.5 的安装文件夹?
标签: python ubuntu cuda tensorflow