【问题标题】:Tensorflow GPU error: ImportError: libcudart.so.7.0: cannot open shared object file: No such file or directoryTensorflow GPU 错误:ImportError: libcudart.so.7.0: cannot open shared object file: No such file or directory
【发布时间】:2016-02-17 23:19:31
【问题描述】:

这是回溯,

>>> import tensorflow
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/__init__.py", line 23, in <module>
    from tensorflow.python import *
  File "/home/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 49, in <module>
    from tensorflow import contrib
  File "/home/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/contrib/__init__.py", line 23, in <module>
    from tensorflow.contrib import layers
  File "/home/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/contrib/layers/__init__.py", line 68, in <module>
    from tensorflow.contrib.layers.python.layers import *
  File "/home/eldor4do/tensorflow_gpu/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/eldor4do/tensorflow_gpu/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/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/python/ops/random_ops.py", line 23, in <module>
    from tensorflow.python.framework import ops
  File "/home/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 39, in <module>
    from tensorflow.python.framework import versions
  File "/home/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/python/framework/versions.py", line 22, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "/home/eldor4do/tensorflow_gpu/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module>
    _pywrap_tensorflow = swig_import_helper()
  File "/home/eldor4do/tensorflow_gpu/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.0: cannot open shared object file: No such file or directory

这是我的 .bashrc 文件,

export CUDA_HOME=/usr/local/cuda 
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64 

PATH=${CUDA_HOME}/bin:${PATH} 
export PATH 

我使用的是 Ubuntu 15.04、cuda7.5 和 cudnn 4.0.7,并且我在 virtualenv 中使用 pip 安装了 tensorflow。

【问题讨论】:

    标签: tensorflow


    【解决方案1】:

    你的问题是cuda7.5。据我了解,TensorFlow 的链接方式仅适用于 cuda7.0。您要么需要从源代码构建它,要么将本地 cuda 版本降级到 7.0(作为一个选项,创建一个 docker 容器并在那里安装较低版本的 cuda,如果您需要 7.5 用于其他目的)。

    【讨论】:

      猜你喜欢
      • 2020-11-05
      • 2018-05-22
      • 2020-04-10
      • 2017-08-12
      • 2020-10-07
      • 1970-01-01
      • 1970-01-01
      • 2021-10-01
      • 1970-01-01
      相关资源
      最近更新 更多