From cdf8f441613cbb9f699d640744218f2b0b3b09d7 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 17 Mar 2023 11:10:37 +0100 Subject: [PATCH] Use dash 2.8.1 for now --- docker/transformers-all-latest-gpu/Dockerfile | 3 +++ docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile | 3 +++ docker/transformers-pytorch-gpu/Dockerfile | 3 +++ 3 files changed, 9 insertions(+) diff --git a/docker/transformers-all-latest-gpu/Dockerfile b/docker/transformers-all-latest-gpu/Dockerfile index 9f30f4531b0ac..da2ed78bd0805 100644 --- a/docker/transformers-all-latest-gpu/Dockerfile +++ b/docker/transformers-all-latest-gpu/Dockerfile @@ -59,6 +59,9 @@ RUN python3 -m pip install --no-cache-dir decord av==9.2.0 ## For `dinat` model #RUN python3 -m pip install --no-cache-dir natten -f https://shi-labs.com/natten/wheels/$CUDA/ +# dash 2.9.0 has some issue. +RUN python3 -m pip install -U dash==2.8.1 + # 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 diff --git a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile index 2fa1317f82758..09160b96aad6f 100644 --- a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile +++ b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile @@ -40,6 +40,9 @@ RUN python3 -m pip uninstall -y deepspeed # TODO: Find out why test fail. RUN DS_BUILD_CPU_ADAM=1 DS_BUILD_FUSED_ADAM=1 DS_BUILD_AIO=1 DS_BUILD_UTILS=1 python3 -m pip install deepspeed --global-option="build_ext" --global-option="-j8" --no-cache -v --disable-pip-version-check 2>&1 +# dash 2.9.0 has some issue. +RUN python3 -m pip install -U dash==2.8.1 + # 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 diff --git a/docker/transformers-pytorch-gpu/Dockerfile b/docker/transformers-pytorch-gpu/Dockerfile index 24564aca6fae4..a5115890a7630 100644 --- a/docker/transformers-pytorch-gpu/Dockerfile +++ b/docker/transformers-pytorch-gpu/Dockerfile @@ -29,6 +29,9 @@ RUN python3 -m pip uninstall -y tensorflow flax RUN python3 -m pip install --no-cache-dir git+https://github.com/facebookresearch/detectron2.git pytesseract RUN python3 -m pip install -U "itsdangerous<2.1.0" +# dash 2.9.0 has some issue. +RUN python3 -m pip install -U dash==2.8.1 + # 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