Skip to content

Commit b2d53a4

Browse files
committed
Remove PyTorch memory configuration from Dockerfile
1 parent 90254f9 commit b2d53a4

1 file changed

Lines changed: 0 additions & 4 deletions

File tree

docker/Dockerfile

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,6 @@ FROM python:3.11-slim
33
# Opcional: define la ruta de cache huggingface
44
ENV HF_HOME=/cache/huggingface
55

6-
# Configuración de memoria para PyTorch
7-
ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512
8-
ENV MAX_MEMORY=16384
9-
106
# Instala dependencias necesarias
117
RUN apt-get update && apt-get install -y --no-install-recommends \
128
git \

0 commit comments

Comments
 (0)