From 6309c5ae90dbe17c01358d4b86ade3e88855fb70 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Mon, 5 Dec 2022 12:30:57 +0100 Subject: [PATCH] install tensorflow_probability for TF pipeline CI --- 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 f5a38defd749..8c9ca32b9f2e 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