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-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