Spaces:
Sleeping
Sleeping
Fix: writable HF caches + OMP threads + libgomp
Browse files- Dockerfile +12 -0
Dockerfile
CHANGED
|
@@ -4,6 +4,18 @@ ENV DEBIAN_FRONTEND=noninteractive PIP_NO_CACHE_DIR=1
|
|
| 4 |
WORKDIR /app
|
| 5 |
ENV PYTHONPATH=/app/backend:$PYTHONPATH
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
COPY . .
|
| 8 |
|
| 9 |
# Install GPU-ready deps
|
|
|
|
| 4 |
WORKDIR /app
|
| 5 |
ENV PYTHONPATH=/app/backend:$PYTHONPATH
|
| 6 |
|
| 7 |
+
# OpenMP runtime to avoid missing libgomp issues
|
| 8 |
+
RUN apt-get update && apt-get install -y libgomp1 && rm -rf /var/lib/apt/lists/*
|
| 9 |
+
|
| 10 |
+
# Writable caches + sane OMP threads
|
| 11 |
+
ENV HF_HOME=/app/.cache \
|
| 12 |
+
TRANSFORMERS_CACHE=/app/.cache/transformers \
|
| 13 |
+
HUGGINGFACE_HUB_CACHE=/app/.cache/huggingface \
|
| 14 |
+
SENTENCE_TRANSFORMERS_HOME=/app/.cache/sentence-transformers \
|
| 15 |
+
XDG_CACHE_HOME=/app/.cache \
|
| 16 |
+
OMP_NUM_THREADS=1
|
| 17 |
+
RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
|
| 18 |
+
|
| 19 |
COPY . .
|
| 20 |
|
| 21 |
# Install GPU-ready deps
|