Dockerfile

FROM ubuntu:22.04

# 必要なパッケージをインストール
RUN apt-get update && apt-get install -y \
    git wget python3 python3-pip && \
    apt-get clean && rm -rf /var/lib/apt/lists/*
RUN pip install torch==2.3.1 torchvision==0.18.1
# Pythonのデフォルト実行ファイルをpython3に変更
RUN ln -s /usr/bin/python3 /usr/bin/python

# リポジトリをクローン
WORKDIR /app

# app.py をコンテナにコピー
COPY . .
RUN export COMMANDLINE_ARGS="--skip-torch-cuda-test"
# WebUIの起動（app.pyを実行）
WORKDIR /app/stable-diffusion-webui-forge
CMD ["python3", "launch.py", "--share", "--disable-torch-cuda"]