Skip to content

Commit 90254f9

Browse files
committed
Update Dockerfile to increase maximum memory allocation for PyTorch from 8g to 16g
1 parent 6b6e7c2 commit 90254f9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ ENV HF_HOME=/cache/huggingface
55

66
# Configuración de memoria para PyTorch
77
ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512
8-
ENV MAX_MEMORY=8g
8+
ENV MAX_MEMORY=16384
99

1010
# Instala dependencias necesarias
1111
RUN apt-get update && apt-get install -y --no-install-recommends \

0 commit comments

Comments
 (0)