chore: purge pip cache in image

gpustack · Oct 16, 2024 · de9a368 · de9a368
1 parent 964d818
commit de9a368
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -16,13 +16,15 @@ COPY dist/*.whl /tmp/
 
 RUN if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
     echo "Installing x86_64 wheel"; \
-    pip3 install /tmp/*_x86_64.whl vllm; \
+    WHEEL_PACKAGE="$(ls /tmp/*_x86_64.whl)[vllm]"; \
     elif [ "$TARGETPLATFORM" = "linux/arm64" ]; then \
     echo "Installing aarch64 wheel"; \
-    pip3 install /tmp/*_aarch64.whl; \
+    WHEEL_PACKAGE="$(ls /tmp/*_aarch64.whl)"; \
     else \
-    echo "Unsupported platform: $TARGETPLATFORM"; \
-    fi \
-    && rm /tmp/*.whl
+    echo "Unsupported platform: $TARGETPLATFORM" && exit 1; \
+    fi && \
+    pip3 install $WHEEL_PACKAGE &&\
+    rm /tmp/*.whl && \
+    pip3 cache purge
 
 ENTRYPOINT [ "gpustack", "start" ]
diff --git a/hack/build-docker.sh b/hack/build-docker.sh
@@ -0,0 +1,7 @@
+#!/usr/bin/env bash
+
+set -o errexit
+set -o nounset
+set -o pipefail
+
+docker buildx build -t gpustack/gpustack:dev -f Dockerfile --platform linux/amd64,linux/arm64 .