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-gpu/Dockerfile
+++ b/docker/transformers-gpu/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 \
    jupyter \
-    tensorflow \
+    tensorflow==2.9.1 \
    torch

 RUN git clone https://github.com/NVIDIA/apex