Spaces:

soiz1
/

stable-diffusion-web-ui

Runtime error

soiz1 commited on Feb 12

Commit

645915f

verified ·

1 Parent(s): 16b47ef

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -4,7 +4,7 @@ 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 --extra-index-url
 # Pythonのデフォルト実行ファイルをpython3に変更
 RUN ln -s /usr/bin/python3 /usr/bin/python

 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