Skip to content

Commit 1c4de02

Browse files
Dobiasdclaude
andcommitted
Bump TensorFlow to 2.21.0 in the perf-test Dockerfile
Also pin keras==3.14.0 explicitly so the perf-test image matches the versions pinned in CI rather than relying on whatever Keras the TensorFlow wheel happens to pull in. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
1 parent abefac8 commit 1c4de02

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

test/Dockerfile

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,12 @@ RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/bazel-archive-keyring.gp
1414
RUN apt-get update -y && apt-get install -y bazel-6.5.0
1515
RUN ln -s /usr/bin/bazel-6.5.0 /usr/bin/bazel
1616

17-
RUN git clone -b 'v2.18.0' --single-branch --depth 1 https://github.com/tensorflow/tensorflow.git
17+
RUN git clone -b 'v2.21.0' --single-branch --depth 1 https://github.com/tensorflow/tensorflow.git
1818
WORKDIR /tensorflow
1919
RUN ./configure
2020
RUN bazel build -c opt --copt=-march=native --jobs 16 --local_ram_resources=HOST_RAM*.3 -c opt //tensorflow/tools/pip_package:wheel --repo_env=WHEEL_NAME=tensorflow_cpu
21-
RUN pip install --break-system-packages bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow_cpu-2.18.0-cp312-cp312-linux_x86_64.whl
21+
RUN pip install --break-system-packages bazel-bin/tensorflow/tools/pip_package/wheel_house/tensorflow_cpu-2.21.0-cp312-cp312-linux_x86_64.whl
22+
RUN pip install --break-system-packages "keras==3.14.0"
2223
WORKDIR /
2324

2425
RUN git clone -b 'v0.2.24' --single-branch --depth 1 https://github.com/Dobiasd/FunctionalPlus && cd FunctionalPlus && mkdir -p build && cd build && cmake .. && make && make install

0 commit comments

Comments
 (0)