jwphantom
commited on
Commit
•
503c42b
1
Parent(s):
e892487
fix problem get_relevant_documents
Browse files- Dockerfile +4 -0
Dockerfile
CHANGED
@@ -12,6 +12,10 @@ RUN pip wheel --no-cache-dir --no-deps --wheel-dir /app/wheels -r requirements.t
|
|
12 |
# Étape finale
|
13 |
FROM python:3.11
|
14 |
|
|
|
|
|
|
|
|
|
15 |
WORKDIR /app
|
16 |
|
17 |
# Créer le répertoire de cache et définir les permissions
|
|
|
12 |
# Étape finale
|
13 |
FROM python:3.11
|
14 |
|
15 |
+
ENV CUDA_VISIBLE_DEVICES=""
|
16 |
+
ENV TF_FORCE_CPU_ALLOW_GROWTH=true
|
17 |
+
ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:32
|
18 |
+
|
19 |
WORKDIR /app
|
20 |
|
21 |
# Créer le répertoire de cache et définir les permissions
|