Updated dockerfile

Author: djthorpe

Makefile

@@ -51,7 +51,7 @@ docker: docker-dep submodule
 		--build-arg OS=${OS} \
 		--build-arg SOURCE=${BUILD_MODULE} \
 		--build-arg VERSION=${VERSION} \
-		-f etc/Dockerfile.${OS}-${ARCH} .
+		-f etc/Dockerfile .
 
 # Test whisper bindings
 test: generate libwhisper libggml

etc/Dockerfile

@@ -0,0 +1,41 @@
+ARG BASE_TAG=0.0.10-4-g6421fd2
+ARG BASE_DEV_CONTAINER=ghcr.io/mutablelogic/cuda-dev:${BASE_TAG}
+ARG BASE_RUN_CONTAINER=ghcr.io/mutablelogic/cuda-rt:${BASE_TAG}
+ARG CUDA_DOCKER_ARCH=all
+ARG GO_VERSION=1.22.5
+ARG ARCH
+ARG OS
+
+# Setup build container
+FROM ${BASE_DEV_CONTAINER} AS build
+ARG CUDA_DOCKER_ARCH
+ARG GO_VERSION
+ARG ARCH
+ARG OS
+
+RUN apt-get -y install software-properties-common curl \
+ && add-apt-repository -y ppa:ubuntuhandbook1/ffmpeg6 \
+ && apt-get -y update \
+ && apt-get -y install libavcodec-dev libavdevice-dev libavfilter-dev libavutil-dev libswscale-dev libswresample-dev
+
+# Install go
+RUN curl -sL https://golang.org/dl/go${GO_VERSION}.${OS}-${ARCH}.tar.gz | tar -C /usr/local -xz
+ENV PATH=$PATH:/usr/local/go/bin
+
+# Copy source
+WORKDIR /app
+COPY . .
+
+# Make whisper-server
+ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH}
+ENV GGML_CUDA=1
+RUN make -j$(nproc) server
+
+# Setup runtime container
+FROM ${BASE_RUN_CONTAINER} AS runtime
+RUN apt-get -y update && apt-get -y upgrade && apt-get -y install libgomp1
+COPY --from=build --chmod=755 /app/build/whisper /usr/local/bin/whisper
+
+# Expose
+ENTRYPOINT [ "/usr/local/bin/whisper" ]
+CMD [ "server", "--dir=/data" ]