0403c6f47e
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com> Co-authored-by: Matthew Noto <99706358+RandNMR73@users.noreply.github.com> Co-authored-by: XOR-op <17672363+XOR-op@users.noreply.github.com> Co-authored-by: Zhang Peiyuan <42993249+jzhang38@users.noreply.github.com>
77 lines
2.8 KiB
Docker
77 lines
2.8 KiB
Docker
# syntax=docker/dockerfile:1.7
|
|
ARG CUDA_TAG=12.9.1-cudnn-devel-ubuntu22.04
|
|
FROM nvidia/cuda:${CUDA_TAG}
|
|
|
|
ARG BUILD_FASTVIDEO_KERNEL_FROM_SOURCE=0
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
UV_LINK_MODE=copy
|
|
|
|
SHELL ["/bin/bash", "-c"]
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
gcc-11 g++-11 clang-11 \
|
|
make cmake ninja-build pkg-config nasm \
|
|
git curl wget ca-certificates \
|
|
libssl-dev zlib1g-dev \
|
|
&& rm -rf /var/lib/apt/lists/* \
|
|
&& update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 100 \
|
|
--slave /usr/bin/g++ g++ /usr/bin/g++-11
|
|
|
|
ENV CUDA_HOME=/usr/local/cuda-12.9
|
|
ENV PATH=/root/.local/bin:/opt/venv/bin:${CUDA_HOME}/bin:${PATH}
|
|
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
|
|
ENV VIRTUAL_ENV=/opt/venv
|
|
|
|
RUN curl -LsSf https://astral.sh/uv/install.sh | sh
|
|
|
|
RUN uv venv --python 3.12 --seed /opt/venv \
|
|
&& echo 'source /opt/venv/bin/activate' >> /root/.bashrc
|
|
|
|
WORKDIR /opt/FastVideo
|
|
|
|
COPY . /opt/FastVideo
|
|
|
|
RUN source /opt/venv/bin/activate \
|
|
&& uv pip install --no-cache-dir "/opt/FastVideo[dreamverse]"
|
|
|
|
# Standard docker build does not expose GPUs, while fastvideo-kernel/build.sh
|
|
# detects the CUDA architecture with torch at build time. The FastVideo package
|
|
# install above brings in the pinned fastvideo-kernel package; rebuild from the
|
|
# copied source only on hosts configured for build-time GPU access.
|
|
RUN if [[ "${BUILD_FASTVIDEO_KERNEL_FROM_SOURCE}" == "1" ]]; then \
|
|
sed -i 's/^git submodule update --init --recursive$/if [[ -d ..\/.git ]]; then git submodule update --init --recursive; fi/' \
|
|
/opt/FastVideo/fastvideo-kernel/build.sh \
|
|
&& source /opt/venv/bin/activate \
|
|
&& cd /opt/FastVideo/fastvideo-kernel \
|
|
&& ./build.sh; \
|
|
else \
|
|
echo "Skipping source fastvideo-kernel build; using installed fastvideo-kernel package."; \
|
|
fi
|
|
|
|
# The monorepo ffmpeg installer force-selects conda compiler triplets for
|
|
# local dev shells. Inside this image we explicitly opt into the system
|
|
# gcc/g++ toolchain.
|
|
RUN source /opt/venv/bin/activate \
|
|
&& INSTALL_PREFIX=/opt/ffmpeg-native \
|
|
SOURCE_DIR=/tmp/ffmpeg-native-src \
|
|
FFMPEG_NATIVE_CC=/usr/bin/gcc \
|
|
FFMPEG_NATIVE_CXX=/usr/bin/g++ \
|
|
bash /opt/FastVideo/apps/dreamverse/scripts/install_native_ffmpeg.sh
|
|
|
|
ENV FASTVIDEO_DREAMVERSE_HOME=/var/lib/dreamverse \
|
|
STREAM_MODE=av_fmp4 \
|
|
FASTVIDEO_ENABLE_PROMPT_SAFETY=0 \
|
|
HF_HOME=/root/.cache/huggingface
|
|
|
|
RUN mkdir -p /var/lib/dreamverse
|
|
|
|
EXPOSE 8009
|
|
|
|
HEALTHCHECK --interval=30s --timeout=5s --start-period=2700s --retries=3 \
|
|
CMD curl -fsS http://127.0.0.1:8009/healthz || exit 1
|
|
|
|
ENTRYPOINT ["/opt/FastVideo/apps/dreamverse/docker/docker_entrypoint.sh"]
|
|
CMD ["dreamverse-server", "--host", "0.0.0.0", "--port", "8009"]
|