MogensR commited on
Commit
5aff2d4
·
1 Parent(s): 55df3a4

Update docker/Dockerfile.cpu

Browse files
Files changed (1) hide show
  1. docker/Dockerfile.cpu +4 -3
docker/Dockerfile.cpu CHANGED
@@ -2,14 +2,15 @@
2
  # CPU-only Dockerfile for BackgroundFX Pro
3
  # For development, testing, or CPU-only deployments
4
  # ============================================================================
5
-
6
  FROM python:3.10-slim
7
 
 
8
  ENV PYTHONDONTWRITEBYTECODE=1 \
9
  PYTHONUNBUFFERED=1 \
10
  DEBIAN_FRONTEND=noninteractive \
11
  DEVICE=cpu \
12
- OMP_NUM_THREADS=4
 
13
 
14
  # Install system dependencies
15
  RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -49,4 +50,4 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
49
  EXPOSE 7860 8000
50
 
51
  # Launch
52
- CMD ["python", "-u", "app.py"]
 
2
  # CPU-only Dockerfile for BackgroundFX Pro
3
  # For development, testing, or CPU-only deployments
4
  # ============================================================================
 
5
  FROM python:3.10-slim
6
 
7
+ # Set environment variables - INCLUDING MKL_NUM_THREADS to fix libgomp error
8
  ENV PYTHONDONTWRITEBYTECODE=1 \
9
  PYTHONUNBUFFERED=1 \
10
  DEBIAN_FRONTEND=noninteractive \
11
  DEVICE=cpu \
12
+ OMP_NUM_THREADS=4 \
13
+ MKL_NUM_THREADS=4
14
 
15
  # Install system dependencies
16
  RUN apt-get update && apt-get install -y --no-install-recommends \
 
50
  EXPOSE 7860 8000
51
 
52
  # Launch
53
+ CMD ["python", "-u", "app.py"]