Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/rgryta/stable-diffusion-wsl2-docker
One-click install for StabilityAI's Stable-Diffusion with AUTOMATIC1111's webui
https://github.com/rgryta/stable-diffusion-wsl2-docker
ai docker dockerfile nvidia stability-ai stable-diffusion ubuntu wsl wsl2
Last synced: about 2 months ago
JSON representation
One-click install for StabilityAI's Stable-Diffusion with AUTOMATIC1111's webui
- Host: GitHub
- URL: https://github.com/rgryta/stable-diffusion-wsl2-docker
- Owner: rgryta
- License: mit
- Created: 2022-12-12T15:51:24.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-02-11T09:48:09.000Z (11 months ago)
- Last Synced: 2024-02-11T10:31:43.172Z (11 months ago)
- Topics: ai, docker, dockerfile, nvidia, stability-ai, stable-diffusion, ubuntu, wsl, wsl2
- Language: PowerShell
- Homepage:
- Size: 31.3 KB
- Stars: 92
- Watchers: 4
- Forks: 13
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Stable Diffusion on Docker (WSL2)
## About
This repository is meant to allow for easy installation of Stable Diffusion on Windows. One click to install. Second click to start.
This setup is completely dependant on current versions of AUTOMATIC1111's webui repository and StabilityAI's Stable-Diffusion models.
In it's current configuration only Nvidia GPUs are supported. This script will also install all dependencies (including xformers) in order to speed up launching the webui.
## Prerequisites
Before following through with these instructions. Please verify below.
1. You have virtualization support - easiest way is to check if you can see "Virtualization" section in Windows Task Manager -> Performance -> CPU (it's located under "More details" if you don't see the Performance tab).
1. You have virtualization enabled - you have to enable it in your BIOS if you don't.
1. You have Windows 11 Pro - you can also use Windows 11 Home (also Windows 10 above certain version), but I cannot guarantee that provided scripts will work their magic.
1. You have Nvidia GPU - this is mandatory for current configuration. Support for AMD is presumably possible, but won't be added until such request shows up. Make sure you also have the newest drivers! Whole repository is based on CUDA 12 - you will be limited to GTX 900-series or higher.
1. You need admin access. These scripts use a PowerShell library that I've prepared, called [WSLTools](https://github.com/rgryta/PowerShell-WSLTools) (handles automatic and interactive installation of WSL distributions from source), you need to have admin privileges to install this module.
## How to use
After installation simply execute start.bat file to start the Stable-Diffusion app. You can open it under [http://localhost:7860](http://localhost:7860).
If you want to close the app - simply launch stop.bat, it will terminate the application and close the terminals.
Note! Keep in mind that stop.bat will terminate and remove all containers based on Stable-Diffusion webui image. If you have downloaded additional models while the application was running - e.g. GAN models - they will have to be redownloaded again.
## Installation
### Automatic
Run install.bat in order to install the Stable Diffusion. This will take a while - as long as you don't see red errors - everything's fine. I takes about 20min to install on my machine.
### Manual
1. Install Windows 11
1. Install WSL from MS Store (https://www.microsoft.com/store/productId/9P9TQF7MRM4R)
1. Search for "Turn Windows features on or off" and enable "Hyper-V"
1. Set WSL to use v2: `wsl --set-default-version 2`
1. Install Linux distro of your choice (Ubuntu given as example): `wsl --install Ubuntu`
1. Set up your username and password
1. (In distro command line) `sudo sh -c 'echo "[boot]\nsystemd=true" > /etc/wsl.conf'`
1. Check your distro name using `wsl --list`
1. Shutdown all distros `wsl --shutdown` and restart the one we're using `wsl --distribution Ubuntu`
1. Make sure you have nvidia drivers installed on Windows
1. Now open WSL. From now on, everything is executed from there.
1. Execute following scripts (installs cuda drivers):
```bash
sudo apt-key del 7fa2af80
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/3bf863cc.pub
sudo add-apt-repository 'deb https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/ /'
sudo apt-get update
sudo apt-get -y install cuda
```
1. Check if you're able to see your GPU in WSL: nvidia-smi
1. Install docker:
```bash
curl https://get.docker.com | sh \
&& sudo systemctl --now enable docker
```
1. Prepare gpg keys to install nvidia-docker:
```bash
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
```
1. Now we can install it: `sudo apt-get install -y nvidia-docker2`
1. Restart docker service: `sudo systemctl restart docker`
1. Check if docker container also sees your GPU: `sudo docker run --rm --gpus all nvidia/cuda:12.0.1-base-ubuntu22.04 nvidia-smi`
1. Run `./build.sh` from repo directory to build the container. You can uncomment depth, upscaler, inpainting and gfpgan from Dockerfile (first generated image) but it will take much more space - default installation is ~25GB total.
1. Run `./run.sh` to start container. Open http://localhost:7860 to access the webui - you can do so from Windows of course.
## Sources
1. [StabilityAI Stable-Diffusion GitHub](https://github.com/Stability-AI/stablediffusion)
1. [StabilityAI Stable-Diffusion HuggingFace](https://huggingface.co/stabilityai/stable-diffusion-2-1)
1. [AUTOMATIC1111 webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
1. [Nvidia Container Runtime](https://nvidia.github.io/nvidia-container-runtime/)
1. [Ubuntu GPU acceleration on WSL2](https://ubuntu.com/tutorials/enabling-gpu-acceleration-on-ubuntu-on-wsl2-with-the-nvidia-cuda-platform#3-install-nvidia-cuda-on-ubuntu)
1. [MS WSL systemd](https://devblogs.microsoft.com/commandline/systemd-support-is-now-available-in-wsl/)
1. [Nvidia WSL](https://docs.nvidia.com/cuda/wsl-user-guide/index.html)