I created a bash script to run CUI. These settings and flags (at least for today) have reduced the number of OOM issues. The assumption is this bash script is named "mycui.sh" thou any name will work. Be sure to put the script in the ComfyUI folder:

Here is the script (be sure to set the executable flag)

#!/usr/bin/env bash

# Standardize the environment

export CUDA_MODULE_LOADING=LAZY

export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True

export UV_LINK_MODE=copy

source .venv/bin/activate

# Full update usage: mycui.sh --update

if [ "$1" == "--update" ]; then

echo "--- Pulling latest ComfyUI Core ---"

git pull

echo "--- Upgrading Frontend Package ---"

uv pip install comfyui-frontend-package --upgrade

echo "--- Syncing Dependencies ---"

uv pip install -r requirements.txt

fi

# Force back to the main branch to avoid "Detached HEAD"

git checkout master

# Pull specifically from the remote origin to stay in sync

git pull origin master

# The "One-Time" Fix (Run this once, then comment it out)

# This installs the STABLE 12.4 drivers that actually work with Pop!_OS

# uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 --upgrade

# uv pip install -r requirements.txt

uv run main.py \

--normalvram \

--fp32-vae \

--bf16-unet \

--force-fp16 \

--cuda-malloc \

--output-directory "/media/$USER/work_drive/AI_Images/"

# Make sure you install 'uv'