From 0be978e19476617f7a3a242928872cfdb3ed50cb Mon Sep 17 00:00:00 2001 From: nerdylive123 <70305325+nerdylive123@users.noreply.github.com> Date: Wed, 17 Apr 2024 01:08:58 +0700 Subject: [PATCH 1/2] Add LD_PRELOAD TMALLOC ( memory problem fix ) --- official-templates/fast-stable-diffusion/Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/official-templates/fast-stable-diffusion/Dockerfile b/official-templates/fast-stable-diffusion/Dockerfile index b78e0b5..2815cf6 100644 --- a/official-templates/fast-stable-diffusion/Dockerfile +++ b/official-templates/fast-stable-diffusion/Dockerfile @@ -8,6 +8,7 @@ WORKDIR / ENV SHELL=/bin/bash \ PYTHONUNBUFFERED=1 \ DEBIAN_FRONTEND=noninteractive + LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc_minimal.so.4 # Set up system RUN apt-get update --yes && \ @@ -44,6 +45,8 @@ RUN wget https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependenci https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependencies/rename_1.10-1_all.deb \ https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependencies/zip_3.0-11build1_amd64.deb \ https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependencies/unzip_6.0-25ubuntu1.1_amd64.deb && \ + https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependencies/libtcmalloc-minimal4_2.5-2.2ubuntu3_amd64.deb && \ + https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dependencies/libgoogle-perftools-dev_2.5-2.2ubuntu3_amd64.deb && \ apt update && \ apt install -y --no-install-recommends libfontconfig1 libgles2-mesa-dev && \ dpkg -i *.deb && \ From db6de486501299cfafce3b0e6afb1167b4e45366 Mon Sep 17 00:00:00 2001 From: nerdylive123 <70305325+nerdylive123@users.noreply.github.com> Date: Wed, 17 Apr 2024 02:26:20 +0700 Subject: [PATCH 2/2] add slash --- official-templates/fast-stable-diffusion/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/official-templates/fast-stable-diffusion/Dockerfile b/official-templates/fast-stable-diffusion/Dockerfile index 2815cf6..df74b29 100644 --- a/official-templates/fast-stable-diffusion/Dockerfile +++ b/official-templates/fast-stable-diffusion/Dockerfile @@ -7,7 +7,7 @@ FROM nvidia/cuda:11.8.0-devel-ubuntu22.04 WORKDIR / ENV SHELL=/bin/bash \ PYTHONUNBUFFERED=1 \ - DEBIAN_FRONTEND=noninteractive + DEBIAN_FRONTEND=noninteractive \ LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc_minimal.so.4 # Set up system