CLMARRARA commited on
Commit
b0fef3b
·
1 Parent(s): 10eed77

Tentativa de ignorar GPU no HF

Browse files
Files changed (3) hide show
  1. Dockerfile +6 -0
  2. app/api.py +1 -1
  3. app/bootstrap.py +12 -0
Dockerfile CHANGED
@@ -15,8 +15,14 @@ ENV MEDIAPIPE_DISABLE_GPU=1
15
  # Configura o EGL para usar o modo surfaceless, evitando problemas de renderização em ambientes sem GPU
16
  ENV EGL_PLATFORM=surfaceless
17
 
 
18
  ENV CUDA_VISIBLE_DEVICES=-1
19
 
 
 
 
 
 
20
  ENV TRANSFORMERS_CACHE=/tmp
21
 
22
  ENV HF_HOME=/tmp
 
15
  # Configura o EGL para usar o modo surfaceless, evitando problemas de renderização em ambientes sem GPU
16
  ENV EGL_PLATFORM=surfaceless
17
 
18
+ # Desativa o uso da GPU para o TensorFlow (se aplicável)
19
  ENV CUDA_VISIBLE_DEVICES=-1
20
 
21
+ # Reduz o nível de log do TensorFlow para evitar mensagens de aviso desnecessárias
22
+ ENV TF_CPP_MIN_LOG_LEVEL=3
23
+
24
+ ENV TRANSFORMERS_OFFLINE=0
25
+
26
  ENV TRANSFORMERS_CACHE=/tmp
27
 
28
  ENV HF_HOME=/tmp
app/api.py CHANGED
@@ -1,9 +1,9 @@
 
1
  from fastapi import FastAPI, UploadFile, File, Form, Request
2
  from fastapi.middleware.cors import CORSMiddleware
3
  import shutil
4
  import os
5
  import json
6
- import app.bootstrap
7
  from app.config import JSON_OUTPUT_PATH, VIDEO_TEMP_PATH, LOG_FILE
8
  from app.pipeline import run_pipeline
9
  from app.decision import classify
 
1
+ import app.bootstrap
2
  from fastapi import FastAPI, UploadFile, File, Form, Request
3
  from fastapi.middleware.cors import CORSMiddleware
4
  import shutil
5
  import os
6
  import json
 
7
  from app.config import JSON_OUTPUT_PATH, VIDEO_TEMP_PATH, LOG_FILE
8
  from app.pipeline import run_pipeline
9
  from app.decision import classify
app/bootstrap.py CHANGED
@@ -7,3 +7,15 @@ log(f"DEEPFACE_HOME ajustado para: {os.environ['DEEPFACE_HOME']}")
7
 
8
  os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
9
  log(f"YOLO_CONFIG_DIR ajustado para: {os.environ['YOLO_CONFIG_DIR']}")
 
 
 
 
 
 
 
 
 
 
 
 
 
7
 
8
  os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
9
  log(f"YOLO_CONFIG_DIR ajustado para: {os.environ['YOLO_CONFIG_DIR']}")
10
+
11
+ os.environ["MEDIAPIPE_DISABLE_GPU"] = "1"
12
+ log("MEDIAPIPE_DISABLE_GPU ajustado para: 1")
13
+
14
+ os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
15
+ log("CUDA_VISIBLE_DEVICES ajustado para: -1")
16
+
17
+ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
18
+ log("TF_CPP_MIN_LOG_LEVEL ajustado para: 3")
19
+
20
+ os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
21
+ log("TF_ENABLE_ONEDNN_OPTS ajustado para: 0")