From 91182e3a70bb1d8b0fcaa21885b2eb4a227ad889 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 5 Dec 2022 16:07:25 +0100 Subject: [PATCH] Install `tensorflow_probability` for TF pipeline CI (#20586) Co-authored-by: ydshieh --- docker/transformers-tensorflow-gpu/Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docker/transformers-tensorflow-gpu/Dockerfile b/docker/transformers-tensorflow-gpu/Dockerfile index f5a38defd7..8c9ca32b9f 100644 --- a/docker/transformers-tensorflow-gpu/Dockerfile +++ b/docker/transformers-tensorflow-gpu/Dockerfile @@ -18,6 +18,8 @@ RUN [ ${#TENSORFLOW} -gt 0 ] && VERSION='tensorflow=='$TENSORFLOW'.*' || VERSIO RUN python3 -m pip uninstall -y torch flax RUN python3 -m pip install -U "itsdangerous<2.1.0" +RUN python3 -m pip install --no-cache-dir -U tensorflow_probability + # When installing in editable mode, `transformers` is not recognized as a package. # this line must be added in order for python to be aware of transformers. RUN cd transformers && python3 setup.py develop