firer1946/iopaint-gpu repository overview

ROM nvidia/cuda:11.8.0-base-ubuntu22.04

RUN apt-get update && apt-get install -y --no-install-recommends \
    software-properties-common \
    libsm6 libxext6 ffmpeg libfontconfig1 libxrender1 libgl1-mesa-glx \
    curl python3-pip

RUN pip3 install --upgrade pip
RUN pip3 install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu118
RUN pip3 install iopaint

EXPOSE 8080

CMD ["bash"]

version: '3.8'

services:
  iopaint:
    image: firer1946/iopaint-gpu
    command: /usr/local/bin/iopaint start --model=lama --device=cuda --port=8080 --host=0.0.0.0
    ports:
      - "8080:8080"
    volumes:
      - ./torch-cache:/root/.cache/torch
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]