利用bazel 可以根据不同平台,编译tensorflow源码,打包成so库,在不同平台上用。
编译时不同tensorflow版本与bazel的版本都需要一一对应。可进行以下方法找到并且切换bazel版本。
1、下载tensorflow源码,git clone https://github.com/tensorflow/tensorflow.gitGitHub - tensorflow/tensorflow: An Open Source Machine Learning Framework for Everyone
2、cd 进去tensorflow ,有bazel的,直接bazel clean。如果不符合,会显示相对应需要的bazel版本,如果执行成功,则证明当前bazel版本可以直接编译。
3、如果步骤2中,没有安装bazel,去网址以下网址根据需求下载,可下载
https://github.com/bazelbuild/bazel/tags
国内源:Index of bazel-local (huaweicloud.com)
这里建议下载sh的文件,安装较为 方便;
4、下载后,chmod a+x ***.sh文件 ,然后安装
sudo ./bazel***.sh --user
5、安装完成后可看到会有一个bin的东西安装在某个路径,具体可看安装完成后的信息。然后添加路径上述的路径到系统内。 export PATH=/home/******/bin/:$PATH
6、最后查询 bazel --version即可。
7、如果想更换tensorflow ,用其他版本的bazel编译,只需要重复2-4的步骤即可。更新后的bazel会覆盖原来路径/home/******/bin/的版本,即可直接用于编译