docker : Add dockerfile for cublas (#1286)

* Create Dockerfile * Rename Dockerfile to cublas.Dockerfile * Rename cublas.Dockerfile to .devops/cublas.Dockerfile --------- Co-authored-by: bobqianic <[email protected]>
ggerganov · Oct 11, 2023 · a0040f5 · a0040f5
1 parent 940cdb1
commit a0040f5
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/.devops/cublas.Dockerfile 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"]