soiz1 commited on
Commit
758cab0
·
verified ·
1 Parent(s): 8b31674

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -2,7 +2,7 @@ FROM ubuntu:22.04
2
 
3
  # 必要なパッケージをインストール
4
  RUN apt-get update && apt-get install -y \
5
- git wget python3 python3-pip && \
6
  apt-get clean && rm -rf /var/lib/apt/lists/*
7
  RUN pip install torch==2.3.1 torchvision==0.18.1
8
  # Pythonのデフォルト実行ファイルをpython3に変更
@@ -18,10 +18,10 @@ COPY . .
18
  RUN export COMMANDLINE_ARGS="--skip-torch-cuda-test"
19
 
20
  RUN mkdir -p ~/.cache/pip
21
- RUN chmod 777 ~/.cache/pip
22
 
23
  RUN mkdir -p ~/.local
24
- RUN chmod 777 ~/.local
25
  # WebUIの起動(app.pyを実行)
26
  WORKDIR /app/stable-diffusion-webui-forge
27
  CMD ["python3", "launch.py", "--share", "--disable-torch-cuda", "--skip-torch-cuda-test"]
 
2
 
3
  # 必要なパッケージをインストール
4
  RUN apt-get update && apt-get install -y \
5
+ git wget python3 python3-pip sudo && \
6
  apt-get clean && rm -rf /var/lib/apt/lists/*
7
  RUN pip install torch==2.3.1 torchvision==0.18.1
8
  # Pythonのデフォルト実行ファイルをpython3に変更
 
18
  RUN export COMMANDLINE_ARGS="--skip-torch-cuda-test"
19
 
20
  RUN mkdir -p ~/.cache/pip
21
+ RUN chmod 777 ~/.cache/pip && sudo chown -R $(whoami):$(whoami) ~/.local
22
 
23
  RUN mkdir -p ~/.local
24
+ RUN chmod 777 ~/.local && sudo chown -R $(whoami):$(whoami) ~/.cache/pip
25
  # WebUIの起動(app.pyを実行)
26
  WORKDIR /app/stable-diffusion-webui-forge
27
  CMD ["python3", "launch.py", "--share", "--disable-torch-cuda", "--skip-torch-cuda-test"]