一 安装 bazel
方法1:
1.下载 bazel
https://github.com/bazelbuild/bazel/releases/tag/5.0.0-pre.20210604.6
2.将 bazel-5.0.0-pre.20210604.6-installer-linux-x86_64.sh 放入ubuntu中,执行:
chmod 777 bazel-5.0.0-pre.20210604.6-installer-linux-x86_64.sh
./bazel-5.0.0-pre.20210604.6-installer-linux-x86_64.sh --user
3.设定环境变量,打开:
vim /etc/profile
在最后添加:
#set bazel environment
export BAZEL_HOME=/root
export PATH=$BAZEL_HOME:$PATH
执行:
source /etc/profile
4.查看bazel 版本:
bazel --version
方法2:
1.下载并执行bazel:
#cd "/usr/local/lib/bazel/bin" && wget https://releases.bazel.build/3.7.2/release/bazel3.7.2-linux-x86_64 && chmod +x bazel-3.7.2-linux-x86_64
wget https://releases.bazel.build/3.7.2/release/bazel-3.7.2-linux-x86_64
chmod +x bazel-3.7.2-linux-x86_64
./bazel-3.7.2-linux-x86_64
2. 设定环境变量,打开:
vim /etc/profile
在最后添加:
#set bazel environment
export BAZEL_HOME=/tensorflow2.0/bazel-3.7.2-linux-x86_64
export PATH=$BAZEL_HOME:$PATH
执行:
source /etc/profile
4.查看bazel 版本:
bazel --version
二 安装 jdk
1.下载jdk:
sudo apt-get install openjdk-8-jdk
2.查看版本:
java -version
3.添加环境变量:
export JAVA_HOME=/usr/lib/jvm/java-8-openjdk-amd64
export JRE_HOME=$JAVA_HOME/jre
export CLASSPATH=.:$JAVA_HOME/lib:$JRE_HOME/lib:$JAVA_HOME/lib/tools.jar
export PATH=$PATH:$JAVA_HOME/bin:$JRE_HOME/bin
4.下载ndk:
https://developer.android.google.cn/ndk/downloads/index.html
5. 放入ubuntu,进行解压
unzip android-ndk-r21e-linux-x86_64.zip
6.设定环境变量,打开
vim /etc/profile
在最后添加:
#set android ndk environment
export ANDROID_NDK_HOME=/tensorflow2.0/android-ndk-r21e
export PATH=$ANDROID_NDK_HOME:$PATH
执行:
source /etc/profile
三 编译tensorflow
1.下载tensorflow
git clone https://github.com/tensorflow/tensorflow.git
2.执行:
./configure
3.编译:
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package