From 2272324fd610eaee079a0b5073916aea4620b282 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9bastien=20Prud=27homme?= Date: Mon, 29 May 2023 23:12:27 +0200 Subject: [PATCH] feat: add CuBLAS support in Docker images (#403) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Sébastien Prud'homme --- Dockerfile | 42 ++++++++++++++++++++++++++--- Dockerfile.dev | 71 +++++++++++++++++++++++++++++++++++++++++++++----- 2 files changed, 103 insertions(+), 10 deletions(-) diff --git a/Dockerfile b/Dockerfile index 1f3830d..d730d4b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,15 +1,49 @@ ARG GO_VERSION=1.20 -ARG BUILD_TYPE= 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 REBUILD=true + WORKDIR /build -RUN apt-get update && apt-get install -y cmake curl libgomp1 libopenblas-dev libopenblas-base libopencv-dev libopencv-core-dev libopencv-core4.5 ca-certificates + +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 ln -s /usr/include/opencv4/opencv2/ /usr/include/opencv2 RUN make build -ENV HEALTHCHECK_ENDPOINT=http://localhost:8080/readyz + # Define the health check command HEALTHCHECK --interval=30s --timeout=360s --retries=10 \ CMD curl -f $HEALTHCHECK_ENDPOINT || exit 1 + EXPOSE 8080 ENTRYPOINT [ "/build/entrypoint.sh" ] diff --git a/Dockerfile.dev b/Dockerfile.dev index 1e355f1..df44359 100644 --- a/Dockerfile.dev +++ b/Dockerfile.dev @@ -1,20 +1,79 @@ ARG GO_VERSION=1.20 ARG DEBIAN_VERSION=11 +FROM golang:$GO_VERSION as builder + 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} -FROM golang:$GO_VERSION as builder WORKDIR /build -RUN apt-get update && apt-get install -y cmake libgomp1 libopenblas-dev libopenblas-base libopencv-dev libopencv-core-dev libopencv-core4.5 -RUN ln -s /usr/include/opencv4/opencv2/ /usr/include/opencv2 + +RUN apt-get update && \ + apt-get install -y cmake + +# 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 build FROM debian:$DEBIAN_VERSION -COPY --from=builder /build/local-ai /usr/bin/local-ai -RUN apt-get update && apt-get install -y ca-certificates curl + +ARG BUILD_TYPE= +ARG GO_TAGS= +ARG CUDA_MAJOR_VERSION=11 +ARG CUDA_MINOR_VERSION=7 + +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 + +RUN apt-get update && \ + apt-get install -y ca-certificates curl + +# CuBLAS requirements +RUN if [ "${BUILD_TYPE}" = "cublas" ]; then \ + apt-get install -y curl 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-cudart-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcublas-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \ + ; fi + +# OpenBLAS requirements +RUN apt-get install -y libopenblas0 + +# Stable Diffusion requirements +RUN apt-get install -y libgomp1 libopencv-core4.5 libopencv-imgcodecs4.5 + +COPY --from=builder /build/local-ai /usr/bin/local-ai + # Define the health check command HEALTHCHECK --interval=30s --timeout=360s --retries=10 \ CMD curl -f $HEALTHCHECK_ENDPOINT || exit 1 + EXPOSE 8080 -ENTRYPOINT [ "/usr/bin/local-ai" ] \ No newline at end of file +ENTRYPOINT [ "/usr/bin/local-ai" ]