pin TF 2.9.1 for self-hosted CIs (#18925)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-09-07 19:46:14 +02:00
parent 2ef7742117
commit 6690ba3f4d
5 changed files with 5 additions and 5 deletions
--- a/docker/transformers-tensorflow-cpu/Dockerfile
+++ b/docker/transformers-tensorflow-cpu/Dockerfile
@@ -15,7 +15,7 @@ RUN apt update && \
 RUN python3 -m pip install --no-cache-dir --upgrade pip && \
    python3 -m pip install --no-cache-dir \
    mkl \
-    tensorflow-cpu
+    tensorflow-cpu==2.9.1

 WORKDIR /workspace
 COPY . transformers/