Sign inSign up

enderkiller5532/forge-webui-nvidia

By enderkiller5532

•Updated 7 months ago

SD Forge WebUI Docker image with NVIDIA Blackwell support, includes docker-compose and NixOS manual.

Image
0

1.3K

enderkiller5532/forge-webui-nvidia repository overview

All info for this image. https://github.com/Enderkiller5532/sd-webui-nvidia-docker-nixos#⁠

⁠Install nvidia-container-toolkit to your system.(For NixOS/Arch users see, the end of page)
docker run \
  --name stable-diffusion-forge \
  --device nvidia.com/gpu=all \
  -e NVIDIA_VISIBLE_DEVICES=all \
  -e NVIDIA_DRIVER_CAPABILITIES=compute,utility \
  -p 7860:7860 \
  -v "$(pwd)/mnt/outputs:/mnt/outputs" \
  -v "$HOME/ai/forge/models/checkpoints:/app/forge/models/Stable-diffusion" \
  -v "$HOME/ai/forge/models/loras:/app/forge/models/Lora" \
  --shm-size=8g \
  forge-web-ui-nvidia:latest

Then you can use: docker start stable-diffusion-forge # or use (-d) if you don't want to see logs and keep the terminal active

Docker-compose:

#version: '3.8'
services:
  forge-ui:
    image: enderkiller5532/forge-webui-nvidia:latest
    container_name: stable-diffusion-forge
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    ports:
      - "127.0.0.1:7860:7860"
    volumes:
      - ./mnt/outputs:/mnt/outputs
      - ~/ai/forge/models/checkpoints:/app/forge/models/Stable-diffusion
      - ~/ai/forge/models/loras:/app/forge/models/Lora
    # NixOS GPU - not only for nixos
    devices:
      - nvidia.com/gpu=all
    shm_size: '8gb'

docker-compose -d up

⁠NixOS⁠

    virtualisation.docker.enable = true;  
    #virtualisation.docker.daemon.settings
    users.extraGroups.docker.members = [ "username-with-access-to-socket(Enderkiller5532)" ];
    hardware.nvidia-container-toolkit.enable = true;

⁠Arch⁠

sudo pacman -S nvidia-container-toolkit

Remember first it fail,then it start normally. GithubRepo⁠ NixOS_Wiki⁠

Tag summary

Content type

Image

Digest

sha256:ca13bf3b7…

Size

10.4 GB

Last updated

7 months ago

docker pull enderkiller5532/forge-webui-nvidia