aducsdr commited on
Commit
3381b5d
·
verified ·
1 Parent(s): 02be4ba

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -11
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- # Dockerfile (VERSÃO FINAL COM A CORREÇÃO 'yes')
2
 
3
  # 1. COMEÇAR COM A BASE CORRETA
4
  FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
@@ -19,33 +19,35 @@ RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86
19
  # 4. ADICIONAR CONDA AO PATH
20
  ENV PATH /opt/conda/bin:$PATH
21
 
22
- # 5. !!! PASSO DE CORREÇÃO FINAL !!!
23
- # Usamos o comando 'yes' para responder "sim" automaticamente à pergunta dos Termos de Serviço.
24
  RUN yes | conda tos accept
25
 
26
- # 6. (OPCIONAL, MAS RECOMENDADO) Agora que os ToS foram aceitos, atualizamos o Conda.
27
  RUN conda update -n base -c defaults conda
28
 
29
- # 7. CRIAR O AMBIENTE CONDA A PARTIR DO ARQUIVO
30
  COPY environment.yml .
31
  RUN conda env create -f environment.yml
32
 
33
- # 8. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
 
 
 
34
  RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
35
 
36
- # 9. DEFINIR O AMBIENTE CONDA COMO PADRÃO
37
  SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
38
 
39
- # 10. PREPARAR O APLICATIVO
40
  WORKDIR /app
41
  RUN git clone https://github.com/bytedance-seed/SeedVR.git
42
  WORKDIR /app/SeedVR
43
 
44
- # 11. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
45
  RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
46
 
47
- # 12. COPIAR O CÓDIGO DO NOSSO APP
48
  COPY app.py .
49
 
50
- # 13. DEFINIR O COMANDO DE EXECUÇÃO
51
  CMD ["python", "app.py"]
 
1
+ # Dockerfile (VERSÃO FINAL COM LIMPEZA DE CACHE)
2
 
3
  # 1. COMEÇAR COM A BASE CORRETA
4
  FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
19
  # 4. ADICIONAR CONDA AO PATH
20
  ENV PATH /opt/conda/bin:$PATH
21
 
22
+ # 5. ACEITAR OS TERMOS DE SERVIÇO
 
23
  RUN yes | conda tos accept
24
 
25
+ # 6. ATUALIZAR O CONDA
26
  RUN conda update -n base -c defaults conda
27
 
28
+ # 7. CRIAR O AMBIENTE CONDA
29
  COPY environment.yml .
30
  RUN conda env create -f environment.yml
31
 
32
+ # 8. !!! PASSO DE OTIMIZAÇÃO !!! LIMPAR CACHES PARA LIBERAR MEMÓRIA
33
+ RUN conda clean --all -y
34
+
35
+ # 9. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
36
  RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
37
 
38
+ # 10. DEFINIR O AMBIENTE CONDA COMO PADRÃO
39
  SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
40
 
41
+ # 11. PREPARAR O APLICATIVO
42
  WORKDIR /app
43
  RUN git clone https://github.com/bytedance-seed/SeedVR.git
44
  WORKDIR /app/SeedVR
45
 
46
+ # 12. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
47
  RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
48
 
49
+ # 13. COPIAR O CÓDIGO DO NOSSO APP
50
  COPY app.py .
51
 
52
+ # 14. DEFINIR O COMANDO DE EXECUÇÃO
53
  CMD ["python", "app.py"]