Commit c32ab44f authored by nanahira's avatar nanahira

update cuda and requirements

parent b08e12d7
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 as base FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 as base
RUN apt update \ RUN apt update \
&& apt-key adv --fetch-keys \ && apt-key adv --fetch-keys \
...@@ -8,7 +8,7 @@ RUN apt update \ ...@@ -8,7 +8,7 @@ RUN apt update \
RUN useradd -ms /bin/bash sd RUN useradd -ms /bin/bash sd
USER sd USER sd
RUN pip install --no-cache-dir torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 RUN pip install --no-cache-dir torch==2.1.2 torchvision==0.16.2 --extra-index-url https://download.pytorch.org/whl/cu121
RUN mkdir /home/sd/stable-diffusion-webui RUN mkdir /home/sd/stable-diffusion-webui
WORKDIR /home/sd/stable-diffusion-webui WORKDIR /home/sd/stable-diffusion-webui
...@@ -49,7 +49,7 @@ RUN mkdir repositories && \ ...@@ -49,7 +49,7 @@ RUN mkdir repositories && \
pip install --no-cache-dir \ pip install --no-cache-dir \
# install requirements of Stable Diffusion # install requirements of Stable Diffusion
transformers==4.19.2 diffusers invisible-watermark triton \ transformers==4.30.2 diffusers invisible-watermark triton \
# install k-diffusion # install k-diffusion
git+https://github.com/crowsonkb/k-diffusion.git \ git+https://github.com/crowsonkb/k-diffusion.git \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment