官方Windows平台编译文档
前期准备
不适用于Linux系统!!!!!!!!!
硬件
软件
请自备科学上网工具,部分依赖安装时会从google网盘下载
- VS2019(VC及Windows SDK)
- Bazel 3.1.0(Bazel为Google官方的Cmake替代品)
- MSYS2
- python 3.6.7(官网下载的本地安装包,非微软应用商店版本)
- TensorFlow 源码(v2.4.2)
C盘预留20~30G硬盘空间,因为Bazel会在C盘用户目录下保存依赖缓存
编译步骤
以下所有指令均在Windows PowerShell执行
1.安装python依赖
pip install six numpy wheel
pip install keras_applications==1.0.6 --no-deps
pip install keras_preprocessing==1.0.5 --no-deps
2. 下载并添加Bazel到环境变量
下载地址:bazel 3.1.0
- 下载完成后,解压到本地自己喜欢存放的位置(压缩包只有一个.exe文件),并将其重命名为bazel.exe。
- 将其路径添加到环境变量->系统变量->Path里
系统路径 环境变量
3. 安装MSYS2并添加环境变量
官方下载: 地址
- 将安装目录下的\usr\bin目录添加到系统变量Path里。假设我本地安装路径为E:\msys64,对应环境变量设置如下图
- 打开powershell输入以下指令
pacman -S git patch unzip
4. 添加Bazel_VS与BAZEL_VC_FULL_VERSION环境变量
这两个必须一起填,否则会报找不到VC的错误,BAZEL_VC_FULL_VERSION设置自己的VC版本,可以进入\VC\Tools里的文件夹查看当前有哪些版本的VC,如果没有请使用VIsual Studio Installer安装。
5. 克隆TensorFlow源码
- 在powershell进入到源码储存空间后,执行以下指令
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
git checkout v2.4.2
6. 编译配置
- 进入tensorflow源码根目录,执行以下指令进行配置
configure.cmd
python configure.py
- 脚本一个七个步骤根据自己需求选择即可
CUDA版本请自行根据官方教程探索,本文章只是用于TensorFlow Lite编译的前期环境测试
7. 执行编译命令
在tensorflow源码根目录执行以下指令,并确保能连通外网
bazel build //tensorflow/tools/pip_package:build_pip_package --define=no_tensorflow_py_deps=true --local_ram_resources=4096 --cxxopt=-std=c++11
8. 编译成功
如果提示404 time out,可重新执行编译指令
问题合集
1. 构建python .whl软件包
请自行查看官方步骤: 构建软件包
|