Move to uv in dstack-server Docker image (#2509)

Author: r4victor

.github/workflows/build.yml

@@ -28,10 +28,11 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - name: Set up uv
+        uses: astral-sh/setup-uv@v5
         with:
           python-version: 3.11
-      - run: python -m pip install pre-commit
+      - run: uv tool install pre-commit
       - run: pre-commit run -a --show-diff-on-failure
 
   frontend-build:

.github/workflows/release.yml

@@ -19,10 +19,11 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - name: Set up uv
+        uses: astral-sh/setup-uv@v5
         with:
           python-version: 3.11
-      - run: python -m pip install pre-commit
+      - run: uv tool install pre-commit
       - run: pre-commit run -a --show-diff-on-failure
 
   frontend-build:

docker/server/release/Dockerfile

@@ -11,13 +11,18 @@ RUN apt-get update && apt-get install -y \
     curl \
     software-properties-common \
     git \
+    sqlite3 \
     && rm -rf /var/lib/apt/lists/*
 
 RUN if [ $(uname -m) = "aarch64" ]; then ARCH="arm64"; else ARCH="amd64"; fi && \
     curl https://github.com/benbjohnson/litestream/releases/download/v0.3.9/litestream-v0.3.9-linux-$ARCH.deb -O -L && \
     dpkg -i litestream-v0.3.9-linux-$ARCH.deb
 
-RUN pip install "dstack[all]==$VERSION" --progress-bar off
+ADD https://astral.sh/uv/install.sh /uv-installer.sh
+RUN sh /uv-installer.sh && rm /uv-installer.sh
+ENV PATH="/root/.local/bin/:$PATH" 
+
+RUN uv tool install "dstack[all]==$VERSION"
 
 COPY entrypoint.sh entrypoint.sh
 RUN chmod 777 entrypoint.sh

docker/server/stgn/Dockerfile

@@ -9,15 +9,20 @@ RUN apt-get update && apt-get install -y \
     curl \
     software-properties-common \
     git \
+    sqlite3 \
     && rm -rf /var/lib/apt/lists/*
 
 RUN if [ $(uname -m) = "aarch64" ]; then ARCH="arm64"; else ARCH="amd64"; fi && \
     curl https://github.com/benbjohnson/litestream/releases/download/v0.3.9/litestream-v0.3.9-linux-$ARCH.deb -O -L && \
     dpkg -i litestream-v0.3.9-linux-$ARCH.deb
 
+ADD https://astral.sh/uv/install.sh /uv-installer.sh
+RUN sh /uv-installer.sh && rm /uv-installer.sh
+ENV PATH="/root/.local/bin/:$PATH"
+
+COPY pyproject.toml uv.lock README.md ./
 COPY src src
-COPY setup.py README.md ./
-RUN pip install '.[all]'
+RUN SETUPTOOLS_SCM_PRETEND_VERSION=0.0.0 uv sync --extra all
 
 COPY docker/server/entrypoint.sh entrypoint.sh
 RUN chmod 777 entrypoint.sh