Skip to content

Commit 2ccf1ae

Browse files
Dobiasdclaude
andcommitted
Update perf-test Dockerfile to build TF 2.21 from source
TF 2.21's .bazelrc uses --enable_workspace, which requires Bazel 7+, so bump the pinned bazel from 6.5.0 to 7.7.0 (matches TF 2.21's .bazelversion). Also install xxd, which TF's XLA genrules now invoke during the build. Verified by running docker build of test/Dockerfile end-to-end: TF wheel builds, save_application_examples.py converts all Keras Application models, applications_performance loads and benchmarks them. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
1 parent 1c4de02 commit 2ccf1ae

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

test/Dockerfile

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,13 @@ FROM ubuntu:24.04
66
RUN apt-get update -y --fix-missing
77
ENV DEBIAN_FRONTEND=noninteractive
88
RUN TZ=Etc/UTC apt-get -y install tzdata
9-
RUN apt-get install -y build-essential apt-utils cmake python3 python3-pip git llvm-17 clang-17 apt-transport-https curl gnupg patchelf
9+
RUN apt-get install -y build-essential apt-utils cmake python3 python3-pip git llvm-17 clang-17 apt-transport-https curl gnupg patchelf xxd
1010

1111
RUN curl -fsSL https://bazel.build/bazel-release.pub.gpg | gpg --dearmor >bazel-archive-keyring.gpg
1212
RUN mv bazel-archive-keyring.gpg /usr/share/keyrings
1313
RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/bazel-archive-keyring.gpg] https://storage.googleapis.com/bazel-apt stable jdk1.8" | tee /etc/apt/sources.list.d/bazel.list
14-
RUN apt-get update -y && apt-get install -y bazel-6.5.0
15-
RUN ln -s /usr/bin/bazel-6.5.0 /usr/bin/bazel
14+
RUN apt-get update -y && apt-get install -y bazel-7.7.0
15+
RUN ln -s /usr/bin/bazel-7.7.0 /usr/bin/bazel
1616

1717
RUN git clone -b 'v2.21.0' --single-branch --depth 1 https://github.com/tensorflow/tensorflow.git
1818
WORKDIR /tensorflow

0 commit comments

Comments
 (0)