Tensorflow on GPU

[Benedikt Boecking]

Have you tried setting the cuda environment path?

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

> On Jun 30, 2016, at 4:31 PM, Hanqi Sun <hanqis at andrew.cmu.edu> wrote:
> 
> Hi all,
> 
> I am a research assistant for Prof. Poczos and would like to use Tensorflow for my project.
> 
> When I try to execute
> 
> import tensorflow
> 
> on the GPU node, I get the following error message
> 
> Traceback (most recent call last):
>   File "<stdin>", line 1, in <module>
>   File "/usr/lib/python2.7/site-packages/tensorflow/__init__.py", line 23, in <module>
>     from tensorflow.python import *
>   File "/usr/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 50, in <module>
>     from tensorflow.python.framework.framework_lib import *
>   File "/usr/lib/python2.7/site-packages/tensorflow/python/framework/framework_lib.py", line 62, in <module>
>     from tensorflow.python.framework.ops import Graph
>   File "/usr/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 40, in <module>
>     from tensorflow.python.framework import versions
>   File "/usr/lib/python2.7/site-packages/tensorflow/python/framework/versions.py", line 24, in <module>
>     from tensorflow.python import pywrap_tensorflow
>   File "/usr/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 28, in <module>
>     _pywrap_tensorflow = swig_import_helper()
>   File "/usr/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
>  
>  
> Could anyone check this out and try to fix it? Thanks!
> 
> Also, I will greatly appreciate it if any of you could share your experience with Tensorflow (especially about image processing) with me as I am new to this framework. If you are interested, please send a private email to me and I am happy to stop by at your office at any time.
> 
> Best,
> Hanqi

-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mailman.srv.cs.cmu.edu/pipermail/autonlab-users/attachments/20160630/3091df8f/attachment.html>