Small fixes to prepare_tcmalloc for Debian/Ubuntu compatibility (d29fe44e) · Commits · github_fork / Stable Diffusion Webui

- /usr/sbin (where ldconfig is usually located) is not typically on users' PATHs by default, so we set that variable before trying to run ldconfig. - The libtcmalloc library is called libtcmalloc_minimal on Debian/Ubuntu systems. We now check whether libtcmalloc_minimal exists when running prepare_tcmalloc.

webui.sh

+1 −1

Original line number	Diff line number	Diff line
		@@ -183,7 +183,7 @@ fi
		# Try using TCMalloc on Linux
		prepare_tcmalloc() {
		if [[ "${OSTYPE}" == "linux"* ]] && [[ -z "${NO_TCMALLOC}" ]] && [[ -z "${LD_PRELOAD}" ]]; then
		TCMALLOC="$(ldconfig -p \| grep -Po "libtcmalloc.so.\d" \| head -n 1)"
		TCMALLOC="$(PATH=/usr/sbin:$PATH ldconfig -p \| grep -Po "libtcmalloc(_minimal\|)\.so\.\d" \| head -n 1)"
		if [[ ! -z "${TCMALLOC}" ]]; then
		echo "Using TCMalloc: ${TCMALLOC}"
		export LD_PRELOAD="${TCMALLOC}"

Admin message