Hello, Predrag.
I got it to work with GPU support. I used conda package manager for most of work, but pip for the actual (custom link) tensorflow installation.
$ python
Python 2.7.12 |Anaconda 4.2.0 (64-bit)| (default, Jul 2 2016, 17:42:40)
[GCC 4.4.7 20120313 (Red Hat 4.4.7-1)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
Anaconda is brought to you by Continuum Analytics.
Please check out:
http://continuum.io/thanks and
https://anaconda.org
>>> import tensorflow as tf
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcurand.so locally
>>> tf.test.is_built_with_cuda()
True
---------------------
Make sure to set the following environmental variables before installing tensorflow. In addition, install with pip by specifying a custom whl location of tensorflow GPU version. I only tested it with python 2.7 on SD7, but I assume it will work with python 3.5 as well.
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda-7.5/targets/x86_64-linux/lib/"
export CUDA_HOME=/usr/local/cuda
You will need to manually install all prerequisite modules (conda installed correct versions, while pip didn't):
$ conda install tensorflow
Fetching package metadata .......
Solving package specifications: ..........
Package plan for installation in environment /usr/local/conda/2.7:
The following packages will be downloaded:
package | build
---------------------------|-----------------
conda-env-2.6.0 | 0 502 B
libprotobuf-3.0.0 | 0 6.0 MB
protobuf-3.0.0 | py27_0 407 KB
pbr-1.10.0 | py27_0 100 KB
mock-2.0.0 | py27_0 96 KB
tensorflow-0.10.0rc0 | np111py27_0 25.0 MB
------------------------------------------------------------
Total: 31.7 MB
The following NEW packages will be INSTALLED:
conda-env: 2.6.0-0
libprotobuf: 3.0.0-0
mock: 2.0.0-py27_0
pbr: 1.10.0-py27_0
protobuf: 3.0.0-py27_0
tensorflow: 0.10.0rc0-np111py27_0
Manually install all required packages (with conda). Do not install the version conda wants you to install (it's only CPU enabled).
Below path can be found at
https://www.tensorflow.org/versions/r0.11/get_started/os_setup.html#pip-installation, so use appropriate whl file.
$ pip install
https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc0-cp27-none-linux_x86_64.whl
That should be it. I got tf.test.is_built_with_cuda() test from the page below:
https://www.tensorflow.org/versions/r0.8/api_docs/python/test.html
Please let me know if you are having any issues.
Thank you,
Asya