mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-08-03 21:51:50 +02:00
c88ffbf9ba
This commit modified the musa docker file to selectively copy directories needed for the container image. This commit also added a step to the docker workflow to free up disk space in attempt to make enough room for the large musa build containers. The motivation for this change is to reduce the size of the container image and try to avoid disk usage issues in CI.
41 lines
1.3 KiB
Docker
41 lines
1.3 KiB
Docker
ARG UBUNTU_VERSION=22.04
|
|
# This needs to generally match the container host's environment.
|
|
ARG MUSA_VERSION=rc4.0.1
|
|
# Target the MUSA build image
|
|
ARG BASE_MUSA_DEV_CONTAINER=mthreads/musa:${MUSA_VERSION}-mudnn-devel-ubuntu${UBUNTU_VERSION}
|
|
# Target the MUSA runtime image
|
|
ARG BASE_MUSA_RUN_CONTAINER=mthreads/musa:${MUSA_VERSION}-mudnn-runtime-ubuntu${UBUNTU_VERSION}
|
|
|
|
FROM ${BASE_MUSA_DEV_CONTAINER} AS build
|
|
WORKDIR /app
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y build-essential libsdl2-dev wget cmake git && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/* /tmp/* /var/tmp/*
|
|
|
|
COPY .. .
|
|
# Enable muBLAS
|
|
RUN make base.en CMAKE_ARGS="-DGGML_MUSA=1"
|
|
|
|
RUN find /app/build -name "*.o" -delete && \
|
|
find /app/build -name "*.a" -delete && \
|
|
rm -rf /app/build/CMakeFiles && \
|
|
rm -rf /app/build/cmake_install.cmake && \
|
|
rm -rf /app/build/_deps
|
|
|
|
FROM ${BASE_MUSA_RUN_CONTAINER} AS runtime
|
|
WORKDIR /app
|
|
|
|
RUN apt-get update && \
|
|
apt-get install -y curl ffmpeg wget cmake git && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/* /tmp/* /var/tmp/*
|
|
|
|
COPY --from=build /app/build/bin /app/build/bin
|
|
COPY --from=build /app/samples /app/samples
|
|
COPY --from=build /app/models /app/models
|
|
|
|
ENV PATH=/app/build/bin:$PATH
|
|
ENTRYPOINT [ "bash", "-c" ]
|