update dockerfile

kookmin-sw · May 19, 2024 · 9233500 · 9233500
1 parent a087502
commit 9233500
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference/template_code/diffusion/Dockerfile.kubernetes_gpu b/inference/template_code/diffusion/Dockerfile.kubernetes_gpu
@@ -6,7 +6,7 @@ COPY requirements_kubernetes_gpu.txt /app/requirements.txt
 
 RUN pip3 install --no-cache-dir -r requirements.txt
 
-RUN apt-get update && apt-get install -y apt-utils wget unzip
+RUN apt-get update && apt-get install -y apt-utils wget unzip && apt-get clean && rm -rf /var/lib/apt/lists/*
 
 COPY kubernetes_app_diffusion.py /app/app.py