Stable Diffusion

The Projekt Stable Diffusion web UI by AUTOMATIC1111 is a great way to start experimenting with stable diffusion. I run the following code on a Windows 10 notebook with nvidea

nvidia drivers

It is possible to install nvidia drivers in WSL 2 for Ubuntu 22.04

sudo apt install nvidia-cudnn libcudnn8 libcudnn8-dev
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin 
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
sudo apt-get update

My nvidia-smi output is as follows:

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 515.76.02    Driver Version: 517.48       CUDA Version: 11.7     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  On   | 00000000:01:00.0  On |                  N/A |
| N/A   69C    P0    52W /  N/A |   3741MiB /  4096MiB |     97%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

This is my output for the cuda libs ldconfig -p | grep cuda

libicudata.so.70 (libc6,x86-64) => /lib/x86_64-linux-gnu/libicudata.so.70
libcuda.so.1 (libc6,x86-64) => /usr/lib/wsl/lib/libcuda.so.1

Install Stable Diffusion web UI

It is important to export "LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH" before running webui.sh.

Run the following once to get started:

sudo apt update && sudo apt -y full-upgrade && sudo apt -y autoremove
sudo apt install wget git python3 python3-venv  zlib1g
cd ~
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
wget -P "~/stable-diffusion-webui/models/Stable-diffusion" https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt
export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH
bash <(wget -qO- https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui/master/webui.sh)

Once this works, run webui.sh for the next start.

Open locahost port 7860 to get the web-ui.

Example results