Tentativa de ignorar GPU no HF
Browse files- Dockerfile +6 -0
- app/api.py +1 -1
- 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")
|