Centos7 编译Tensorflow + Bazel
在Centos7 RedHat 机器上自己编译优化cpu指令集的tensorflow 版本
由于模型在cpu 上的运行速度不忍直视,gpu机器上显存又不太够,所以试着自己编译优化的tensorflow 试试速度
环境
主要是机器的环境,不同的环境要做对应的修改:
- 系统:
cat /proc/version
.Linux version 3.10.0-1062.4.3.el7.x86_64
- tensorflow 版本 cpu-1.11;
- gcc: 4.8.5;
- java 8;
- python3
安装python及对应的依赖包
// An highlighted block
sudo apt install python3-dev python3-pip # or python-dev python-pip
// 安装 TensorFlow pip 软件包依赖项(如果使用虚拟环境,请省略 --user 参数):
pip3 install -U --user pip six numpy wheel setuptools mock future>=0.17.1
pip3 install -U --user keras_applications==1.0.6 --no-deps
pip3 install -U --user keras_preprocessing==1.0.5 --no-deps
安装 Bazel
这里有个坑,请注意。要想编译tensorflow 首先你得去确定自己想要的版本,然后去安装对应版本的bazel。
到这里查看:链接: Link.
我需要tensorflow 1.11, 所以对应的是用bazel 0.15.0,也许其他版本也行,可以测试下
编译安装,下载