From 15fd39ea0ec4f826deb4cb8f3064004939544d30 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 20 Oct 2022 20:24:39 +0200 Subject: [PATCH] Install tf2onnx dev version (#19755) * pin tf2onnx<=1.12.0 * Install tf2onnx main * Pin to a specific commit Co-authored-by: ydshieh --- docker/transformers-all-latest-gpu/Dockerfile | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/docker/transformers-all-latest-gpu/Dockerfile b/docker/transformers-all-latest-gpu/Dockerfile index 0dd64f86d30..27f0965f5c0 100644 --- a/docker/transformers-all-latest-gpu/Dockerfile +++ b/docker/transformers-all-latest-gpu/Dockerfile @@ -36,6 +36,11 @@ RUN python3 -m pip install --no-cache-dir -U tensorflow RUN python3 -m pip install --no-cache-dir -U tensorflow_probability RUN python3 -m pip uninstall -y flax jax +# To include the change in this commit https://github.com/onnx/tensorflow-onnx/commit/ddca3a5eb2d912f20fe7e0568dd1a3013aee9fa3 +# Otherwise, we get tf2onnx==1.8 (caused by `flatbuffers` version), and some tests fail with `ValueError: from_keras requires input_signature`. +# TODO: remove this line once the conflict is resolved in these libraries. +RUN python3 -m pip install --no-cache-dir git+https://github.com/onnx/tensorflow-onnx.git@ddca3a5eb2d912f20fe7e0568dd1a3013aee9fa3 + # Use installed torch version for `torch-scatter` to avid to deal with PYTORCH='pre'. # If torch is nightly version, the link is likely to be invalid, but the installation falls back to the latest torch-scatter RUN python3 -m pip install --no-cache-dir torch-scatter -f https://data.pyg.org/whl/torch-$(python3 -c "from torch import version; print(version.__version__.split('+')[0])")+$CUDA.html