From 6d71dd7d988934e110cdb26d21d0d419df73d61b Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Tue, 30 May 2023 15:53:37 +0200 Subject: [PATCH] fix: do not build from the same container (#434) Signed-off-by: mudler --- Dockerfile | 50 +++++++++++++++++++++++++++++++++++++++++++++++++- entrypoint.sh | 4 +++- 2 files changed, 52 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 60c3066..2401e95 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,6 @@ ARG GO_VERSION=1.20 -FROM golang:$GO_VERSION + +FROM golang:$GO_VERSION as builder ARG BUILD_TYPE= ARG GO_TAGS= @@ -41,6 +42,53 @@ RUN apt-get install -y libopencv-dev && \ COPY . . RUN make build +FROM golang:$GO_VERSION + +ARG BUILD_TYPE= +ARG GO_TAGS= +ARG CUDA_MAJOR_VERSION=11 +ARG CUDA_MINOR_VERSION=7 + +ENV BUILD_TYPE=${BUILD_TYPE} +ENV GO_TAGS=${GO_TAGS} +ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility +ENV NVIDIA_REQUIRE_CUDA="cuda>=${CUDA_MAJOR_VERSION}.0" +ENV NVIDIA_VISIBLE_DEVICES=all +ENV HEALTHCHECK_ENDPOINT=http://localhost:8080/readyz +ENV HEALTHCHECK_TIMEOUT=10m +ENV HEALTHCHECK_INTERVAL=10m +ENV HEALTHCHECK_RETRIES=10m + +ENV REBUILD=true + +WORKDIR /build + +RUN apt-get update && \ + apt-get install -y ca-certificates cmake curl + +# CuBLAS requirements +RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \ + apt-get install -y software-properties-common && \ + apt-add-repository contrib && \ + curl -O https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb && \ + dpkg -i cuda-keyring_1.0-1_all.deb && \ + rm -f cuda-keyring_1.0-1_all.deb && \ + apt-get update && \ + apt-get install -y cuda-nvcc-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \ + ; fi +ENV PATH /usr/local/cuda/bin:${PATH} + +# OpenBLAS requirements +RUN apt-get install -y libopenblas-dev + +# Stable Diffusion requirements +RUN apt-get install -y libopencv-dev && \ + ln -s /usr/include/opencv4/opencv2 /usr/include/opencv2 + +COPY . . +RUN make prepare-sources +COPY --from=builder /build/local-ai ./ + # Define the health check command HEALTHCHECK --interval=1m --timeout=10m --retries=10 \ CMD curl -f $HEALTHCHECK_ENDPOINT || exit 1 diff --git a/entrypoint.sh b/entrypoint.sh index e7390e5..28c4045 100755 --- a/entrypoint.sh +++ b/entrypoint.sh @@ -1,9 +1,11 @@ #!/bin/bash +set -e cd /build if [ "$REBUILD" != "false" ]; then - make rebuild + rm -rf ./local-ai + make build fi ./local-ai "$@" \ No newline at end of file