2016-03-01 73 views
7

与GPU在Ubuntu 14.04安装TensorFlow。对于Ubuntu/UNIX环境来说,这是相当新颖的东西,所以很多我不明白的东西。当搜索这个问题,我碰到的是无法导入CUDA 7.0实例来的时候,他们已经安装CUDA 7.5(因为TensorFlow不支持7.5)。我安装了CUDA 7.0,但它似乎在寻找7.5 - 为什么会这样?我在我的.bashrc文件中有以下几行:TensorFlow(GPU)libcudart.so.7.5错误 - 不能打开共享对象文件

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-7.0/lib64" 
export CUDA_HOME=/usr/local/cuda-7.0 

但是我仍然看到下面的问题......任何人有任何建议吗?

>>> import tensorflow as tf 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/__init__.py", line 23, in <module> 
    from tensorflow.python import * 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 49, in <module> 
    from tensorflow import contrib 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/contrib/__init__.py", line 23, in <module> 
    from tensorflow.contrib import layers 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/contrib/layers/__init__.py", line 68, in <module> 
    from tensorflow.contrib.layers.python.layers import * 
    File "/home/ford/tensorflow/local/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/ford/tensorflow/local/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/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/ops/random_ops.py", line 23, in <module> 
    from tensorflow.python.framework import ops 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 39, in <module> 
    from tensorflow.python.framework import versions 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/framework/versions.py", line 22, in <module> 
    from tensorflow.python import pywrap_tensorflow 
    File "/home/ford/tensorflow/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module> 
    _pywrap_tensorflow = swig_import_helper() 
    File "/home/ford/tensorflow/local/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 

回答

1

你需要从SOURCE安装tensorflow配置设置。通过pip或easy_install进行安装不起作用。

我使用PIP从0.6.0升级到tensorflow 0.7.1后,同样的问题。

0

我有同样的问题。您是否曾尝试将CUDA 7.5与7一起安装?这工作对我来说,我改变了我的PATH变量包括7.5而不是7.0

+0

我在安装CUDA 7.5的安装问题,以及由于某种原因,我无法找到我应该运行到指定要使用的CUDA版本的配置文件。我可能会尝试安装TensorFlow 0.7.0,因为它默认为7.0我认为。 – Chimi

0

如果您正在使用最新R0.7的二进制文件,在我的经验,他们现在已经内置在默认情况下,支持CUDA 7.5,而... R0.6二进制文件CUDA 7.0

您可以从源代码编译tensorflow与CUDA 7.0工作,而不是工作,看到这里的文档Configure TensorFlow's canonical view of Cuda libraries

$ ./configure 
Please specify the location of python. [Default is /usr/bin/python]: 
Do you wish to build TensorFlow with GPU support? [y/N] y 
GPU support will be enabled for TensorFlow 

Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave 
empty to use system default]: 7.0 

Please specify the location where CUDA 7.0 toolkit is installed. Refer to 
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda 

Please specify the Cudnn version you want to use. [Leave empty to use system 
default]: 4.0.4 

Please specify the location where the cuDNN 4.0.4 library is installed. Refer to 
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cudnn-r4-rc/ 

Please specify a list of comma-separated Cuda compute capabilities you want to 
build with. You can find the compute capability of your device at: 
https://developer.nvidia.com/cuda-gpus. 
Please note that each additional compute capability significantly increases your 
build time and binary size. [Default is: \"3.5,5.2\"]: 3.5 

Setting up Cuda include 
Setting up Cuda lib64 
Setting up Cuda bin 
Setting up Cuda nvvm 
Configuration finished 
+0

是的,我已经看到了这一点,但无法找到该配置文件,以改变其使用的库。我通过安装默认为CUDA 7.0的TensorFlow 0.7.0来“解决”我的问题... – Chimi

+0

配置脚本位于顶级tensorflow源目录中,然后执行git clone https://github.com/ tensorflow/tensorflow /斑点/主/配置 – j314erre

相关问题