qwen3-tts.Dockerfile

ARG UBUNTU_VERSION=24.04
ARG CUDA_VERSION=13.1.0

ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}

FROM ${BASE_CUDA_RUN_CONTAINER} AS build

RUN apt-get update && apt-get install -y curl sox && \
    curl https://repo.anaconda.com/pkgs/misc/gpgkeys/anaconda.asc | gpg --dearmor > conda.gpg && \
    install -o root -g root -m 644 conda.gpg /usr/share/keyrings/conda-archive-keyring.gpg && \
    gpg --keyring /usr/share/keyrings/conda-archive-keyring.gpg --no-default-keyring --fingerprint 34161F5BF5EB1D4BFBBB8F0A8AEB4F8B29D82806 && \
    echo "deb [arch=amd64 signed-by=/usr/share/keyrings/conda-archive-keyring.gpg] https://repo.anaconda.com/pkgs/misc/debrepo/conda stable main" > /etc/apt/sources.list.d/conda.list && \
    apt-get update && apt-get install -y conda

RUN . /opt/conda/etc/profile.d/conda.sh && \
    conda create -n qwen3-tts python=3.12 -y && \
    conda activate qwen3-tts && \
    pip install https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.7.12/flash_attn-2.7.4+cu130torch2.10-cp312-cp312-linux_x86_64.whl && \
    pip install -U torch torchaudio qwen-tts

ENV HF_HOME=/data

ENTRYPOINT ["/opt/conda/envs/qwen3-tts/bin/qwen-tts-demo"]
CMD ["--help"]