Skip to content

Instantly share code, notes, and snippets.

@leosaa
Last active September 14, 2018 20:54
Show Gist options
  • Save leosaa/745bf19bbfbce714cf2e0e5948d86170 to your computer and use it in GitHub Desktop.
Save leosaa/745bf19bbfbce714cf2e0e5948d86170 to your computer and use it in GitHub Desktop.
Installing TensorFlow locally over Python 3.6.4 on RHEL 7.5

Installing TensorFlow locally over Python 3.6.4 on RHEL 7.5

TensorFlow requires:

Set Environment

rpm -Uvh bazel-0.16.1-1.el7.x86_64.rpm
mkdir $HOME/dl
DL=$HOME/dl
cd ${DL}

Compiling TensorFlow

pip3 install --upgrade pip --user
export PATH=~/.local/bin:$PATH
pip3 install wheel six numpy --user
pip3 install --upgrade setuptools --user
export PYTHONPATH=~/.local/lib/python3.6/site-packages
mkdir tmp
git clone https://github.com/tensorflow/tensorflow
cd tensorflow/
git checkout r1.10
./configure
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

or if you have a GPU, you have to build with CUDA support as follow:

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
pip3 install /tmp/tensorflow_pkg/tensorflow-1.10.1-cp36-cp36m-linux_x86_64.whl --user

Test Enviroment

export PYTHONPATH=~/.local/lib/python3.6/site-packages
python3
import tensorflow as tf
print (tf.__version__)
>>>1.10.1
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
>>>b'Hello, TensorFlow!'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment