vibe-voice (latest)

2025-08-31T15:53:39Z

Pull the image from the command line:

docker pull git.ericxliu.me/eric/vibe-voice:latest

Digest

sha256:84d75466bc55f36eccb47503109639bfb596fda220e886dbf18382fea4d151a1

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

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>=12.1 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 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526

ENV NV_CUDA_CUDART_VERSION=12.1.105-1

ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-12-1

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=12.1.1

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-12-1=${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=12.1.1-1

ENV NV_NVTX_VERSION=12.1.105-1

ENV NV_LIBNPP_VERSION=12.1.0.40-1

ENV NV_LIBNPP_PACKAGE=libnpp-12-1=12.1.0.40-1

ENV NV_LIBCUSPARSE_VERSION=12.1.0.106-1

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-1

ENV NV_LIBCUBLAS_VERSION=12.1.3.1-1

ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-1=12.1.3.1-1

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

ENV NV_LIBNCCL_PACKAGE_VERSION=2.17.1-1

ENV NCCL_VERSION=2.17.1-1

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.17.1-1+cuda12.1

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-12-1=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-12-1=${NV_NVTX_VERSION} libcusparse-12-1=${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_CUDA_LIB_VERSION=12.1.1-1

ENV NV_CUDA_CUDART_DEV_VERSION=12.1.105-1

ENV NV_NVML_DEV_VERSION=12.1.105-1

ENV NV_LIBCUSPARSE_DEV_VERSION=12.1.0.106-1

ENV NV_LIBNPP_DEV_VERSION=12.1.0.40-1

ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-1=12.1.0.40-1

ENV NV_LIBCUBLAS_DEV_VERSION=12.1.3.1-1

ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-1

ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-1=12.1.3.1-1

ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.1.1-1

ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-1=12.1.1-1

ENV NV_NVPROF_VERSION=12.1.105-1

ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-1=12.1.105-1

ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev

ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.17.1-1

ENV NCCL_VERSION=2.17.1-1

ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.17.1-1+cuda12.1

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-cudart-dev-12-1=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-12-1=${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-1=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-12-1=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-12-1=${NV_NVML_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-12-1=${NV_LIBCUSPARSE_DEV_VERSION} ${NV_LIBCUBLAS_DEV_PACKAGE} ${NV_LIBNCCL_DEV_PACKAGE} ${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

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

ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs

ENV NV_CUDNN_VERSION=8.9.0.131

ENV NV_CUDNN_PACKAGE_NAME=libcudnn8

ENV NV_CUDNN_PACKAGE=libcudnn8=8.9.0.131-1+cuda12.1

ENV NV_CUDNN_PACKAGE_DEV=libcudnn8-dev=8.9.0.131-1+cuda12.1

ARG TARGETARCH

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

LABEL com.nvidia.cudnn.version=8.9.0.131

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

ARG PYTORCH_VERSION

ARG TRITON_VERSION

ARG TARGETPLATFORM

ARG CUDA_VERSION

LABEL com.nvidia.volumes.needed=nvidia_driver

RUN |4 PYTORCH_VERSION=2.3.1 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.1 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev && rm -rf /var/lib/apt/lists/* # buildkit

COPY /opt/conda /opt/conda # buildkit

RUN |4 PYTORCH_VERSION=2.3.1 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.1 /bin/sh -c if test -n "${TRITON_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends gcc; rm -rf /var/lib/apt/lists/*; fi # buildkit

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

ENV NVIDIA_VISIBLE_DEVICES=all

ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility

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

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

ENV PYTORCH_VERSION=2.3.1

WORKDIR /workspace

WORKDIR /app

ENV DEBIAN_FRONTEND=noninteractive

ENV PYTHONUNBUFFERED=1

ENV MODEL_PATH=microsoft/VibeVoice-1.5B

ENV OUTPUT_DIR=/data/podcast_output

ENV HF_HOME=/data/hf

ENV TRANSFORMERS_CACHE=/data/hf/transformers

ENV HF_HUB_CACHE=/data/hf/hub

ENV XDG_CACHE_HOME=/data/.cache

ENV CUDA_HOME=/usr/local/cuda

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

ENV USE_FLASH_ATTENTION=

ENV ENABLE_API=1

ENV ALLOW_NO_AUTH=1

RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends ffmpeg git curl build-essential ninja-build && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit

COPY . . # buildkit

RUN /bin/sh -c mkdir -p /data/hf /data/hf/transformers /data/hf/hub /data/.cache /data/tmp /data/pip /data/podcast_output /data/outputs # buildkit

ENV TMPDIR=/data/tmp

ENV PIP_CACHE_DIR=/data/pip

RUN /bin/sh -c pip install -e . # buildkit

RUN /bin/sh -c pip install soundfile requests fastapi==0.112.2 uvicorn==0.30.6 pydantic==2.8.2 starlette==0.38.2 # buildkit

VOLUME [/data]

EXPOSE map[7860/tcp:{}]

CMD ["/bin/sh" "-c" "python demo/gradio_demo.py --model_path \"$MODEL_PATH\" --output_dir \"$OUTPUT_DIR\" --share"]

Key	Value
com.nvidia.cudnn.version	8.9.0.131
com.nvidia.volumes.needed	nvidia_driver
maintainer	NVIDIA CORPORATION <cudatools@nvidia.com>
org.opencontainers.image.ref.name	ubuntu
org.opencontainers.image.version	22.04