From 8639cfb4c2890db95f4c79e493bf2296a98236cb Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 5 Dec 2022 15:08:32 +0100 Subject: [PATCH] Install `natten` with CUDA version (#20546) Co-authored-by: ydshieh --- docker/transformers-all-latest-gpu/Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docker/transformers-all-latest-gpu/Dockerfile b/docker/transformers-all-latest-gpu/Dockerfile index 8feea5d039..5429eca68b 100644 --- a/docker/transformers-all-latest-gpu/Dockerfile +++ b/docker/transformers-all-latest-gpu/Dockerfile @@ -56,6 +56,8 @@ RUN python3 -m pip install --no-cache-dir decord # For `dinat` model RUN python3 -m pip install --no-cache-dir natten +RUN python3 -m pip install --no-cache-dir natten -f https://shi-labs.com/natten/wheels/$CUDA/ + # 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