tensorflow/build

TensorFlow SIG Build Docker镜像文档

1. 镜像概述和主要用途

TensorFlow SIG Build Docker镜像是由TensorFlow SIG Build提供的标准化容器，主要用于构建和测试TensorFlow。该镜像通过公开开发流程维护，经Google开发者验证，并发布于Docker Hub仓库tensorflow/build。其核心用途包括：

• 在CI（持续集成）环境中自动化构建和测试TensorFlow
• 供用户本地复现CI构建流程
• 支持TensorFlow OSS DevInfra团队评估构建tf-nightly（夜间版本）

2. 核心功能和特性

2.1 构建环境标准化

基于NVIDIA CUDA容器镜像构建，确保TensorFlow GPU加速版本的构建兼容性。

2.2 灵活的标签策略

• latest标签：持续更新，跟踪TensorFlow master分支，用于构建最新开发版本。
• 版本号标签（如2.16）：对应特定TensorFlow版本，当新TensorFlow分支创建时，标签会“冻结”以长期支持该分支的构建需求。
• 预构建标签：持续构建当前TensorFlow版本 + 1的标签（如当前版本为2.15时，预构建2.16标签），确保新分支创建后可立即使用。

2.3 可信验证

镜像经Google开发者验证，确保构建环境的一致性和可靠性。

3. 使用场景和适用范围

3.1 CI环境构建与测试

适用于TensorFlow官方或第三方CI流水线，自动化执行构建、单元测试及集成测试。

3.2 本地复现CI构建

供开发者在本地环境拉取对应标签的镜像，复现CI环境中的构建过程，定位构建相关问题。

3.3 TensorFlow版本开发支持

针对特定TensorFlow版本（如2.16）的开发，通过版本号标签固化构建环境，避免依赖变动影响构建稳定性。

4. 详细使用方法和配置说明

4.1 获取镜像

镜像托管于Docker Hub，仓库地址为tensorflow/build。可通过docker pull命令拉取：

bash
# 拉取latest标签（对应master分支）
docker pull tensorflow/build:latest

# 拉取特定版本标签（如2.16，对应TensorFlow 2.16版本）
docker pull tensorflow/build:2.16

4.2 标签选择指南

• 开发最新特性：使用latest标签，跟踪TensorFlow master分支。
• 构建特定版本：使用对应版本号标签（如构建TensorFlow 2.16时使用2.16标签）。
• 预准备新版本构建：使用当前版本+1标签（如当前版本为2.15时，2.16标签已预构建完成）。

4.3 运行容器示例

拉取镜像后，可通过docker run启动容器，进入交互式环境进行构建操作：

bash
# 启动latest标签容器并进入bash终端
docker run -it --rm tensorflow/build:latest bash

5. 注意事项

5.1 许可协议

本镜像基于NVIDIA CUDA容器镜像构建，使用前需遵守NVIDIA CUDA镜像的最终用户许可协议（EULA），详情参见https://hub.docker.com/r/nvidia/cuda%E3%80%82

5.2 CUDA环境依赖

如需使用GPU加速构建，需确保宿主环境已正确安装CUDA，具体参考NVIDIA CUDA安装指南。