mail-classify/docker-compose.yml at main · andersonjuniorz/mail-classify · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
services:

  # ========================================================================================
  # ------------------------------------- BACKEND ------------------------------------------
  # ========================================================================================

  backend:
    build:
      context: ./Backend # Diz ao Docker para buscar o Dockerfile no diretorio atual
      dockerfile: Dockerfile

    ports:
      - "5000:5000" # Mapeia a porta 5000 do seu host (maquina local) para a porta 5000 do container
                    # Isso permite acessar o Flask via http://localhost:5000

    volumes:
      - ./Backend/myApp/data/datasets:/app/myApp/data/datasets # persiste os datasets de treino
      - ./Backend/fine_tuned_classifier:/app/fine_tuned_classifier # persiste a IA mais recente

    env_file:
      - ./Backend/.env

    restart: on-failure


    # ================================================================================
    # --- ATENCAO: ESTA CONFIGURACAO HABILITA ACESSO A GPU NO DOCKER -----------------
    # --- Porem, para GPU's menos potente sera necessario usar a CPU em vez da GPU ---
    # --- para treinar/inferir o modelo DistilBERT. Voce provavelmente vera ----------
    # --- erros de 'CUDA out of memory'. Mantenha USE_GPU=False no seu .env ----------
    # --- para que o treinamento ocorra na CPU sem problemas -------------------------
    # ================================================================================

    # ------ COMENTE ESSE BLOCO E ALTERE USE_GPU=FALSE PARA USAR APENAS A CPU ---------
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all # Tenta usar todas as GPUs NVIDIA disponiveis
              capabilities: [gpu] # Habilita capacidades de GPU


  # ========================================================================================
  # ------------------------------------- FRONTEND -----------------------------------------
  # ========================================================================================

  frontend:
    build:
      context: ./Frontend
      dockerfile: Dockerfile

    ports:
      - "80:80"

    depends_on:
      - backend

    env_file:
      - ./Frontend/.env