我想在我的系统上使用tensorflow-gpu.我重新安装了很多次,它给出了下面的错误.但是当我使用tensorflow-cpu时它工作正常.我安装了cuda 8.0工具包和cudnn 5.1.
Traceback (most recent call last):
File "finetune.py", line 17, in <module>
import tensorflow as tf
File "/home/saurabh/code/env/local/lib/python2.7/site-packages/tensorflow/__init__.py", line 24, in <module>
from tensorflow.python import *
File "/home/saurabh/code/env/local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 72, in <module>
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "/home/saurabh/code/env/local/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 61, in <module>
from tensorflow.python import pywrap_tensorflow
File "/home/saurabh/code/env/local/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module>
_pywrap_tensorflow = swig_import_helper()
File "/home/saurabh/code/env/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.8.0: cannot open shared object file: No such file or directory
Failed to load the native TensorFlow runtime.
See https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md#import_error
for some common reasons and solutions. Include the entire stack trace
above this error message when asking for help.
Run Code Online (Sandbox Code Playgroud)
如果更改路径变量不起作用,在终端中运行以下内容通常可以解决这个问题:
sudo ldconfig /usr/local/cuda/lib64
Run Code Online (Sandbox Code Playgroud)
(有时cuda可能需要类似的东西cuda-8.0,具体取决于您的安装)
问题是您没有CUDA_HOME设置环境变量。如果你认为你已经安装了 CUDA,那么你需要确保上面的环境变量设置为 prob=perly 并且包含所有共享库所在的 CUDA 安装目录的地址。
export CUDA_HOME=/CUDA/installation/path
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64"
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
11047 次 |
| 最近记录: |