whisperx-asr-service

Pull the image from the command line:

docker pull git.flexiblyrigid.au/kstenhouse/whisperx-asr-service@sha256:1b2a0e5245150414b1b11bad721f571a994dcdf0bc7672e6a9d6b51e99a6b4f2

Digest

sha256:1b2a0e5245150414b1b11bad721f571a994dcdf0bc7672e6a9d6b51e99a6b4f2

For more information on the Container registry, see the documentation.

WhisperX ASR - CUDA 11.8 sm_60 Pascal compatible

ARG RELEASE

ARG LAUNCHPAD_BUILD_ARCH

LABEL org.opencontainers.image.ref.name=ubuntu

LABEL org.opencontainers.image.version=22.04

ADD file:63d5ab3ef0aab308c0e71cb67292c5467f60deafa9b0418cbb220affcd078444 in /

CMD ["/bin/bash"]

ENV NVARCH=x86_64

ENV NVIDIA_REQUIRE_CUDA=cuda>=11.8 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471

ENV NV_CUDA_CUDART_VERSION=11.8.89-1

ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-11-8

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates && curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/cuda-keyring_1.0-1_all.deb && dpkg -i cuda-keyring_1.0-1_all.deb && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit

ENV CUDA_VERSION=11.8.0

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-11-8=${NV_CUDA_CUDART_VERSION} ${NV_CUDA_COMPAT_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit

ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64

COPY NGC-DL-CONTAINER-LICENSE / # buildkit

ENV NVIDIA_VISIBLE_DEVICES=all

ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility

ENV NV_CUDA_LIB_VERSION=11.8.0-1

ENV NV_NVTX_VERSION=11.8.86-1

ENV NV_LIBNPP_VERSION=11.8.0.86-1

ENV NV_LIBNPP_PACKAGE=libnpp-11-8=11.8.0.86-1

ENV NV_LIBCUSPARSE_VERSION=11.7.5.86-1

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-11-8

ENV NV_LIBCUBLAS_VERSION=11.11.3.6-1

ENV NV_LIBCUBLAS_PACKAGE=libcublas-11-8=11.11.3.6-1

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

ENV NV_LIBNCCL_PACKAGE_VERSION=2.15.5-1

ENV NCCL_VERSION=2.15.5-1

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.15.5-1+cuda11.8

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-libraries-11-8=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-11-8=${NV_NVTX_VERSION} libcusparse-11-8=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} ${NV_LIBNCCL_PACKAGE_NAME} # buildkit

COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit

COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit

ENV NVIDIA_PRODUCT_NAME=CUDA

ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]

ENV NV_CUDNN_VERSION=8.9.6.50

ENV NV_CUDNN_PACKAGE_NAME=libcudnn8

ENV NV_CUDNN_PACKAGE=libcudnn8=8.9.6.50-1+cuda11.8

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

LABEL com.nvidia.cudnn.version=8.9.6.50

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends ${NV_CUDNN_PACKAGE} && apt-mark hold ${NV_CUDNN_PACKAGE_NAME} && rm -rf /var/lib/apt/lists/* # buildkit

LABEL maintainer=your-name

LABEL description=WhisperX ASR Service — P100/Pascal GPU compatible

LABEL cuda.version=11.8

LABEL torch.version=2.1.2

LABEL compute.capability=sm_60

ENV DEBIAN_FRONTEND=noninteractive

ENV PYTHONUNBUFFERED=1

ENV PYTHONDONTWRITEBYTECODE=1

ENV DEVICE=cuda

ENV COMPUTE_TYPE=float32

ENV BATCH_SIZE=8

ENV PRELOAD_MODEL=medium

ENV MAX_FILE_SIZE_MB=500

RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends python3 python3-pip python3-dev build-essential pkg-config ffmpeg libavformat-dev libavcodec-dev libavdevice-dev libavutil-dev libswscale-dev libswresample-dev git curl ca-certificates && ln -sf /usr/bin/python3 /usr/bin/python && rm -rf /var/lib/apt/lists/* && pip3 install --no-cache-dir --upgrade pip # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir torch==2.1.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118 # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir "numpy<2" # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir "cython<3" && pip3 install --no-cache-dir --no-build-isolation "av==10.0.0" && pip3 install --no-cache-dir --no-deps "faster-whisper==0.10.1" && pip3 install --no-cache-dir "ctranslate2" "requests" && pip3 uninstall -y cython # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir "pyannote.audio==3.1.1" "matplotlib" # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir "transformers<4.47.0" "pandas>=2.2.3" "nltk>=3.9.1" "setuptools>=65" # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir --no-deps "whisperx==3.1.6" # buildkit

COPY patch_asr.py /tmp/patch_asr.py # buildkit

RUN /bin/sh -c python3 /tmp/patch_asr.py && rm /tmp/patch_asr.py # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir "fastapi==0.115.0" "uvicorn[standard]==0.30.6" "python-multipart==0.0.9" # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir "soundfile" "librosa" "huggingface_hub>=0.23.0,<1.0.0" # buildkit

RUN /bin/sh -c pip3 install --no-cache-dir --upgrade "numpy<2" && python3 -c "import numpy; print('numpy', numpy.__version__)" && python3 -c "import torch; print('torch', torch.__version__)" && python3 -c "import faster_whisper; v=faster_whisper.__version__; parts=list(map(int,v.split('.'))); assert parts<[1,0,0], f'faster-whisper {v} too new, need <1.0.0'; print('faster-whisper', v, 'OK')" && python3 -c "import whisperx; print('whisperx import OK')" # buildkit

WORKDIR /workspace

RUN /bin/sh -c git clone --depth 1 --filter=blob:none --sparse https://github.com/murtaza-nasir/whisperx-asr-service.git /tmp/upstream && git -C /tmp/upstream sparse-checkout set app && cp -r /tmp/upstream/app /workspace/app && rm -rf /tmp/upstream # buildkit

EXPOSE [9000/tcp]

HEALTHCHECK &{["CMD-SHELL" "curl -f http://localhost:9000/health || exit 1"] "30s" "10s" "1m0s" "0s" '\x03'}

CMD ["uvicorn" "app.main:app" "--host" "0.0.0.0" "--port" "9000" "--workers" "1"]

Key	Value
com.nvidia.cudnn.version	8.9.6.50
compute.capability	sm_60
cuda.version	11.8
description	WhisperX ASR Service — P100/Pascal GPU compatible
maintainer	your-name
org.opencontainers.image.description	WhisperX ASR - CUDA 11.8 sm_60 Pascal compatible
org.opencontainers.image.ref.name	ubuntu
org.opencontainers.image.revision	d4166d0ce8a6159cadf76fef09ebcefa817ca006
org.opencontainers.image.source	https://git.flexiblyrigid.au/kstenhouse/whisperx-asr-service
org.opencontainers.image.title	WhisperX ASR Service (P100)
org.opencontainers.image.version	22.04
torch.version	2.1.2

whisperx-asr-service (sha256:1b2a0e5245150414b1b11bad721f571a994dcdf0bc7672e6a9d6b51e99a6b4f2)

Installation

About this package

Image layers

Labels