aducsdr commited on
Commit
267743b
·
verified ·
1 Parent(s): 0fd259e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +15 -22
Dockerfile CHANGED
@@ -1,11 +1,9 @@
1
- # Dockerfile (VERSÃO FINAL CORRIGIDA)
2
 
3
- # 1. COMEÇAR COM A BASE CORRETA: Uma imagem oficial da NVIDIA com CUDA 12.1.1 e as ferramentas de desenvolvimento (devel)
4
  FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
5
 
6
  # 2. INSTALAR DEPENDÊNCIAS DO SISTEMA
7
- # Precisamos de wget para baixar o miniconda e git para clonar o repositório.
8
- # DEBIAN_FRONTEND=noninteractive evita que a instalação peça inputs.
9
  ENV DEBIAN_FRONTEND=noninteractive
10
  RUN apt-get update && apt-get install -y --no-install-recommends \
11
  wget \
@@ -13,43 +11,38 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
13
  && apt-get clean \
14
  && rm -rf /var/lib/apt/lists/*
15
 
16
- # 3. INSTALAR O MINICONDA SOBRE A IMAGEM NVIDIA
17
- # Baixamos o instalador, executamos em modo silencioso (-b) no diretório /opt/conda (-p)
18
- # e depois limpamos o instalador.
19
  RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
20
  /bin/bash ~/miniconda.sh -b -p /opt/conda && \
21
  rm ~/miniconda.sh
22
 
23
- # 4. ADICIONAR CONDA AO PATH DO SISTEMA
24
- # Isso garante que os comandos 'conda' e 'pip' usarão a instalação que acabamos de fazer.
25
  ENV PATH /opt/conda/bin:$PATH
26
 
27
- # 5. CRIAR O AMBIENTE CONDA A PARTIR DO NOSSO ARQUIVO
28
- # Copiamos o environment.yml e deixamos o conda instalar as dependências principais.
 
 
 
29
  COPY environment.yml .
30
  RUN conda env create -f environment.yml
31
 
32
- # 6. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
33
- # Agora este comando vai funcionar, pois nvcc e CUDA_HOME estão presentes na imagem base.
34
- # 'conda run -n seedvr ...' executa o comando dentro do nosso ambiente específico.
35
  RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
36
 
37
- # 7. DEFINIR O AMBIENTE CONDA COMO PADRÃO PARA O RESTO DO DOCKERFILE
38
- # Esta é a linha corrigida, em formato JSON válido e completo.
39
  SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
40
 
41
- # 8. PREPARAR O APLICATIVO
42
  WORKDIR /app
43
  RUN git clone https://github.com/bytedance-seed/SeedVR.git
44
  WORKDIR /app/SeedVR
45
 
46
- # 9. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
47
- # O comando huggingface-cli já está disponível no ambiente conda.
48
  RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
49
 
50
- # 10. COPIAR O CÓDIGO DO NOSSO APP
51
  COPY app.py .
52
 
53
- # 11. DEFINIR O COMANDO DE EXECUÇÃO
54
- # O comando 'python' já será o do ambiente 'seedvr'.
55
  CMD ["python", "app.py"]
 
1
+ # Dockerfile (VERSÃO FINAL COM CORREÇÃO DO ToS)
2
 
3
+ # 1. COMEÇAR COM A BASE CORRETA
4
  FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
5
 
6
  # 2. INSTALAR DEPENDÊNCIAS DO SISTEMA
 
 
7
  ENV DEBIAN_FRONTEND=noninteractive
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
  wget \
 
11
  && apt-get clean \
12
  && rm -rf /var/lib/apt/lists/*
13
 
14
+ # 3. INSTALAR O MINICONDA
 
 
15
  RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
16
  /bin/bash ~/miniconda.sh -b -p /opt/conda && \
17
  rm ~/miniconda.sh
18
 
19
+ # 4. ADICIONAR CONDA AO PATH
 
20
  ENV PATH /opt/conda/bin:$PATH
21
 
22
+ # 5. !!! PASSO DE CORREÇÃO !!! ACEITAR OS TERMOS DE SERVIÇO DO CONDA
23
+ # Esta linha resolve o erro CondaToSNonInteractiveError
24
+ RUN conda config --set anaconda_tos_accepted true
25
+
26
+ # 6. CRIAR O AMBIENTE CONDA A PARTIR DO ARQUIVO
27
  COPY environment.yml .
28
  RUN conda env create -f environment.yml
29
 
30
+ # 7. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
 
 
31
  RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
32
 
33
+ # 8. DEFINIR O AMBIENTE CONDA COMO PADRÃO
 
34
  SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
35
 
36
+ # 9. PREPARAR O APLICATIVO
37
  WORKDIR /app
38
  RUN git clone https://github.com/bytedance-seed/SeedVR.git
39
  WORKDIR /app/SeedVR
40
 
41
+ # 10. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
 
42
  RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
43
 
44
+ # 11. COPIAR O CÓDIGO DO NOSSO APP
45
  COPY app.py .
46
 
47
+ # 12. DEFINIR O COMANDO DE EXECUÇÃO
 
48
  CMD ["python", "app.py"]