From: joecryptotoo <redacted>
Date: Wed, 11 Oct 2023 10:00:17 +0000 (-0700)
Subject: docker : Add dockerfile for cublas (#1286)
X-Git-Tag: upstream/1.7.4~1300
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=a0040f5d12bcee7e9e2a77e5b5127904c04698a5;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

docker : Add dockerfile for cublas (#1286)

* Create Dockerfile

* Rename Dockerfile to cublas.Dockerfile

* Rename cublas.Dockerfile to .devops/cublas.Dockerfile

---------

Co-authored-by: bobqianic <redacted>
---

diff --git a/.devops/cublas.Dockerfile b/.devops/cublas.Dockerfile
new file mode 100644
index 00000000..ae3da75f
--- /dev/null
+++ b/.devops/cublas.Dockerfile
@@ -0,0 +1,28 @@
+ARG UBUNTU_VERSION=22.04
+
+# This needs to generally match the container host's environment.
+ARG CUDA_VERSION=11.7.1
+
+# Target the CUDA build image
+ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
+
+FROM ${BASE_CUDA_DEV_CONTAINER} as build
+
+# Unless otherwise specified, we make a fat build.
+ARG CUDA_DOCKER_ARCH=all
+
+RUN apt-get update && \
+    apt-get install -y build-essential git cmake
+
+WORKDIR /app
+
+COPY . .
+
+# Set nvcc architecture
+ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH}
+# Enable cuBLAS
+ENV WHISPER_CUBLAS=1
+
+RUN make
+
+ENTRYPOINT ["/app/main"]