TF: TF 2.10 unpin + related onnx test skips (#18995)

2022-09-12 19:30:27 +01:00
parent 7f4708e1a2
commit 1182b945a6
12 changed files with 33 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==2.9.1 \
+    tensorflow \
    torch

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