From 0322a9734976005166241e2f3823c2fc664bd6ab Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 26 Feb 2024 10:01:33 +0530 Subject: [PATCH] move to uv in the Dockerfiles. --- docker/diffusers-flax-cpu/Dockerfile | 6 +++--- docker/diffusers-flax-tpu/Dockerfile | 8 ++++---- docker/diffusers-onnxruntime-cpu/Dockerfile | 6 +++--- docker/diffusers-onnxruntime-cuda/Dockerfile | 6 +++--- docker/diffusers-pytorch-compile-cuda/Dockerfile | 4 ++-- docker/diffusers-pytorch-cpu/Dockerfile | 6 +++--- docker/diffusers-pytorch-cuda/Dockerfile | 4 ++-- docker/diffusers-pytorch-xformers-cuda/Dockerfile | 4 ++-- 8 files changed, 22 insertions(+), 22 deletions(-) diff --git a/docker/diffusers-flax-cpu/Dockerfile b/docker/diffusers-flax-cpu/Dockerfile index 57a9c1ec7422..ec8aef8cdf3e 100644 --- a/docker/diffusers-flax-cpu/Dockerfile +++ b/docker/diffusers-flax-cpu/Dockerfile @@ -23,13 +23,13 @@ ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) # follow the instructions here: https://cloud.google.com/tpu/docs/run-in-container#train_a_jax_model_in_a_docker_container -RUN python3 -m pip install --no-cache-dir --upgrade pip && \ - python3 -m pip install --upgrade --no-cache-dir \ +RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ + python3 -m uv pip install --upgrade --no-cache-dir \ clu \ "jax[cpu]>=0.2.16,!=0.3.2" \ "flax>=0.4.1" \ "jaxlib>=0.1.65" && \ - python3 -m pip install --no-cache-dir \ + python3 -m uv pip install --no-cache-dir \ accelerate \ datasets \ hf-doc-builder \ diff --git a/docker/diffusers-flax-tpu/Dockerfile b/docker/diffusers-flax-tpu/Dockerfile index 2517da586d74..4b38dd13e461 100644 --- a/docker/diffusers-flax-tpu/Dockerfile +++ b/docker/diffusers-flax-tpu/Dockerfile @@ -23,15 +23,15 @@ ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) # follow the instructions here: https://cloud.google.com/tpu/docs/run-in-container#train_a_jax_model_in_a_docker_container -RUN python3 -m pip install --no-cache-dir --upgrade pip && \ - python3 -m pip install --no-cache-dir \ +RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ + python3 -m uv pip install --no-cache-dir \ "jax[tpu]>=0.2.16,!=0.3.2" \ -f https://storage.googleapis.com/jax-releases/libtpu_releases.html && \ - python3 -m pip install --upgrade --no-cache-dir \ + python3 -m uv pip install --upgrade --no-cache-dir \ clu \ "flax>=0.4.1" \ "jaxlib>=0.1.65" && \ - python3 -m pip install --no-cache-dir \ + python3 -m uv pip install --no-cache-dir \ accelerate \ datasets \ hf-doc-builder \ diff --git a/docker/diffusers-onnxruntime-cpu/Dockerfile b/docker/diffusers-onnxruntime-cpu/Dockerfile index ef8205e28254..6c43e4cde602 100644 --- a/docker/diffusers-onnxruntime-cpu/Dockerfile +++ b/docker/diffusers-onnxruntime-cpu/Dockerfile @@ -22,14 +22,14 @@ RUN python3 -m venv /opt/venv ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) -RUN python3 -m pip install --no-cache-dir --upgrade pip && \ - python3 -m pip install --no-cache-dir \ +RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ + python3 -m uv pip install --no-cache-dir \ torch==2.1.2 \ torchvision==0.16.2 \ torchaudio==2.1.2 \ onnxruntime \ --extra-index-url https://download.pytorch.org/whl/cpu && \ - python3 -m pip install --no-cache-dir \ + python3 -m uv pip install --no-cache-dir \ accelerate \ datasets \ hf-doc-builder \ diff --git a/docker/diffusers-onnxruntime-cuda/Dockerfile b/docker/diffusers-onnxruntime-cuda/Dockerfile index 76db288f59ef..ce59e039dc25 100644 --- a/docker/diffusers-onnxruntime-cuda/Dockerfile +++ b/docker/diffusers-onnxruntime-cuda/Dockerfile @@ -22,14 +22,14 @@ RUN python3 -m venv /opt/venv ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) -RUN python3 -m pip install --no-cache-dir --upgrade pip && \ - python3 -m pip install --no-cache-dir \ +RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ + python3 -m uv pip install --no-cache-dir \ torch==2.1.2 \ torchvision==0.16.2 \ torchaudio==2.1.2 \ "onnxruntime-gpu>=1.13.1" \ --extra-index-url https://download.pytorch.org/whl/cu117 && \ - python3 -m pip install --no-cache-dir \ + python3 -m uv pip install --no-cache-dir \ accelerate \ datasets \ hf-doc-builder \ diff --git a/docker/diffusers-pytorch-compile-cuda/Dockerfile b/docker/diffusers-pytorch-compile-cuda/Dockerfile index 4b3841a57684..c90a441ab7c2 100644 --- a/docker/diffusers-pytorch-compile-cuda/Dockerfile +++ b/docker/diffusers-pytorch-compile-cuda/Dockerfile @@ -24,8 +24,8 @@ RUN python3.9 -m venv /opt/venv ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) -RUN python3.9 -m pip install --no-cache-dir --upgrade pip && \ - python3.9 -m pip install --no-cache-dir \ +RUN python3.9 -m pip install --no-cache-dir --upgrade pip uv && \ + python3.9 -m uv pip install --no-cache-dir \ torch \ torchvision \ torchaudio \ diff --git a/docker/diffusers-pytorch-cpu/Dockerfile b/docker/diffusers-pytorch-cpu/Dockerfile index 127c61a719c5..75473c3069d7 100644 --- a/docker/diffusers-pytorch-cpu/Dockerfile +++ b/docker/diffusers-pytorch-cpu/Dockerfile @@ -23,14 +23,14 @@ RUN python3 -m venv /opt/venv ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) -RUN python3 -m pip install --no-cache-dir --upgrade pip && \ - python3 -m pip install --no-cache-dir \ +RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ + python3 -m uv pip install --no-cache-dir \ torch \ torchvision \ torchaudio \ invisible_watermark \ --extra-index-url https://download.pytorch.org/whl/cpu && \ - python3 -m pip install --no-cache-dir \ + python3 -m uv pip install --no-cache-dir \ accelerate \ datasets \ hf-doc-builder \ diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 7fc3d8ced992..3da6751de9ef 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -23,8 +23,8 @@ RUN python3 -m venv /opt/venv ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) -RUN python3 -m pip install --no-cache-dir --upgrade pip && \ - python3 -m pip install --no-cache-dir \ +RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ + python3 -m uv pip install --no-cache-dir \ torch \ torchvision \ torchaudio \ diff --git a/docker/diffusers-pytorch-xformers-cuda/Dockerfile b/docker/diffusers-pytorch-xformers-cuda/Dockerfile index 8f2619c62394..ca57bebe3318 100644 --- a/docker/diffusers-pytorch-xformers-cuda/Dockerfile +++ b/docker/diffusers-pytorch-xformers-cuda/Dockerfile @@ -23,13 +23,13 @@ RUN python3 -m venv /opt/venv ENV PATH="/opt/venv/bin:$PATH" # pre-install the heavy dependencies (these can later be overridden by the deps from setup.py) -RUN python3 -m pip install --no-cache-dir --upgrade pip && \ +RUN python3 -m pip install --no-cache-dir --upgrade pip uv && \ python3 -m pip install --no-cache-dir \ torch \ torchvision \ torchaudio \ invisible_watermark && \ - python3 -m pip install --no-cache-dir \ + python3 -m uv pip install --no-cache-dir \ accelerate \ datasets \ hf-doc-builder \