Running Tensorflow On Gpu Cluster In Virtualenv

I installed the GPU version of tensorflow in a virtualenv following these instructions. The problem is, I am getting a segmentation fault upon starting a session. That is, this cod

Solution 1:

It's not finding CuDNN -

I tensorflow/stream_executor/dso_loader.cc:93] Couldn't open CUDA library > libcudnn.so.6.5. LD_LIBRARY_PATH: :/vol/cuda/7.0.28/lib64 I tensorflow/stream_executor/cuda/cuda_dnn.cc:1382] Unable to load cuDNN DSO

You need to have it installed. Please see the TensorFlow CUDA installation instructions

Solution 2:

After untar the cudnn

[root@localhost cudnn]# cd include/
[root@localhostinclude]# mv cudnn.h /usr/local/cuda/include/
[root@localhostinclude]# cd ../lib64/
[root@localhost lib64]# mv * /usr/local/cuda/lib

And it is ok

[root@localhost ~]# python
Python 2.7.5 (default, Sep 15 2016, 22:37:39) 
[GCC 4.8.5 20150623 (Red Hat 4.8.5-4)] on linux2
Type "help", "copyright", "credits" or "license"for more information.
>>> import tensorflow as f
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so.5 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so.8.0 locally
>>>