TensorFlow 源码编译安装

## Install prerequisites (rhel)
yum install numpy python-devel python-wheel python-mock ## Install Bazel
wget https://github.com/bazelbuild/bazel/releases/download/0.4.5/bazel-0.4.5-dist.zip
unzip bazel-0.4.-dist.zip -d bazel ## Modify bazel WORKSPACE
cd bazel
Modify WORKSPACE (Optinal step to avoid warnings)
from workspace(name = "io_bazel")
to workspace(name = "bazel_tools") ## Build bazel
./compile.sh ## Install cuda 8.0
https://developer.nvidia.com/cuda-downloads ## Install cuDNN
sudo tar -xvf cudnn-8.0-linux-ppc64le-v5..tgz -C /usr/local/
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170427/cudnn-8.0-linux-ppc64le-v6.0-tgz ## Build TensorFlow
git clone https://github.com/tensorflow/tensorflow
cd tensorflow export PYTHON_BIN_PATH=/usr/bin/python
export GCC_HOST_COMPILER_PATH=/usr/bin/gcc
export TF_NEED_JEMALLOC=
export TF_NEED_GCP=
export TF_NEED_HDFS=
export TF_ENABLE_XLA=
export TF_NEED_CUDA=
export TF_NEED_OPENCL=
export CUDA_TOOLKIT_PATH=/usr/local/cuda-8.0
export TF_CUDA_VERSION=8.0
export TF_CUDNN_VERSION=
export CC_OPT_FLAGS="-O3"
export PATH=$PATH:/home/xxx/bazel/output/ (Add your bazel bin path)

./configure

----------------------------
Build with GPU
----------------------------
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
pip install --user /tmp/tensorflow_pkg/tensorflow-xxx.whl export LD_LIBRARY_PATH=/usr/local/cuda/targets/ppc64le-linux/lib/:$LIBRARY_PATH
export PYTHON_LIBRARY_PATH=/home/xlhu/.local/lib/python2./site-packages/ # Run an example to check
export PYTHON_LIBRARY_PATH=$PYTHON_LIBRARY_PATH:/home/xlhu/.local/lib/python3./site-packages/ (Depends on your own path)
git clone https://github.com/tensorflow/models.git
cd models/tutorials/image/mnist
python convolutional.py ----------------------------
Build with CPU Only
----------------------------
bazel build //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
pip install --user /tmp/tensorflow_pkg/tensorflow-xxx.whl
上一篇:Hexo搭建静态博客踩坑日记(二)


下一篇:JavaScript 新手的踩坑日记