TensorFlow 源码编译安装
2024-09-05 21:32:17
## 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
最新文章
- Entity Framework Code First实体对象变动跟踪
- history
- Python学习资料整理以及书籍、开发工具推荐
- 安装python官方的mysql库“mysql-connector-python”
- 【Shell脚本学习6】Shell变量:Shell变量的定义、删除变量、只读变量、变量类型
- 【CSS】Beginner6:Border
- 类linux系统/proc/sysrq-trigger文件功能作用
- 显示GetLastError()的错误描述字符串
- 完美解决IE(IE6/IE7/IE8)不兼容HTML5标签的方法zt
- RIO包 健壮的I/O函数代码
- zBase --轻量级DOM操作库
- 实验:Oracle数据泵导出导入之序列问题
- 【转】LDA数学八卦
- 【SSH/SFTP】SSH协议和SFTP
- Linux show funny time[ liunx 有趣的东西展示]
- Linux lvs-NAT模式配置详解
- CSS: pseudo-classes and pseudo-elements
- codeforces500B
- 并发之线程封闭与ThreadLocal解析
- poi导出excel合并单元格(包括列合并、行合并)