Update docker/Dockerfile.cpu
Browse files- 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"]
|