Revert "TF: unpin maximum TF version (#18917)" (#18972)

This reverts commit d8cf3b2087.
2022-09-10 09:11:46 -04:00
parent d8cf3b2087
commit a26114777e
7 changed files with 11 additions and 11 deletions
--- a/docker/transformers-tensorflow-cpu/Dockerfile
+++ b/docker/transformers-tensorflow-cpu/Dockerfile
@@ -15,11 +15,11 @@ 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/
 RUN cd transformers/ && \
    python3 -m pip install --no-cache-dir .

-CMD ["/bin/bash"]
+CMD ["/bin/bash"]