SeedVR2 / Dockerfile
aducsdr's picture
Update Dockerfile
383a83f verified
raw
history blame
2.08 kB
# Dockerfile (VERSÃO FINAL COM CAMINHO EXPLÍCITO)
# 1. COMEÇAR COM A BASE CORRETA
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
# 2. INSTALAR DEPENDÊNCIAS DO SISTEMA
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends \
wget \
git \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*
# 3. INSTALAR O MINICONDA
RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
/bin/bash ~/miniconda.sh -b -p /opt/conda && \
rm ~/miniconda.sh
# 4. ADICIONAR CONDA AO PATH (Importante para os comandos RUN)
ENV PATH /opt/conda/bin:$PATH
# 5. ACEITAR OS TERMOS DE SERVIÇO
RUN yes | conda tos accept
# 6. ATUALIZAR O CONDA
RUN conda update -n base -c defaults conda
# 7. CRIAR O AMBIENTE CONDA
COPY environment.yml .
RUN conda env create -f environment.yml && conda clean --all -y
# 8. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
# 9. DEFINIR O SHELL PADRÃO PARA OS COMANDOS 'RUN'
SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
# 10. PREPARAR O APLICATIVO
WORKDIR /app
RUN git clone https://github.com/bytedance-seed/SeedVR.git
WORKDIR /app/SeedVR
# 11. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
# 12. BAIXAR OS VÍDEOS DE EXEMPLO
RUN wget -O 01.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/23_1_lq.mp4 && \
wget -O 02.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/28_1_lq.mp4 && \
wget -O 03.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/2_1_lq.mp4
# 13. COPIAR O CÓDIGO DO NOSSO APP
COPY app.py .
# 14. !!! A CORREÇÃO FINAL E EXPLÍCITA !!!
# Executa o app.py usando o caminho absoluto para o Python do nosso ambiente Conda.
CMD ["/opt/conda/envs/seedvr/bin/python", "app.py"]