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 \
&& apt-key adv --fetch-keys \
......@@ -8,7 +8,7 @@ RUN apt update \
RUN useradd -ms /bin/bash 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
WORKDIR /home/sd/stable-diffusion-webui
......@@ -49,7 +49,7 @@ RUN mkdir repositories && \
pip install --no-cache-dir \
# install requirements of Stable Diffusion
transformers==4.19.2 diffusers invisible-watermark triton \
transformers==4.30.2 diffusers invisible-watermark triton \
# install k-diffusion
git+https://github.com/crowsonkb/k-diffusion.git \
......
GitPython
Pillow
accelerate
blendmodes
clean-fid
diskcache
einops
facexlib
fastapi>=0.90.1
GitPython==3.1.32
Pillow==9.5.0
accelerate==0.21.0
blendmodes==2022
clean-fid==0.1.35
diskcache==5.6.3
einops==0.4.1
facexlib==0.3.0
fastapi==0.94.0
pydantic==1.10.8
gfpgan
gradio==3.41.2
inflection
jsonmerge
kornia
lark
numpy
omegaconf
open-clip-torch
piexif
inflection==0.5.1
jsonmerge==1.8.0
kornia==0.6.7
lark==1.1.2
numpy==1.26.2
omegaconf==2.2.3
open-clip-torch==2.20.0
piexif==1.1.3
protobuf==3.20.0
psutil
psutil==5.9.5
pytorch_lightning==1.9.4
realesrgan
requests
resize-right
safetensors
scikit-image>=0.19
tomesd
resize-right==0.0.2
safetensors==0.4.2
scikit-image==0.21.0
tomesd==0.1.3
torch
torchdiffeq
torchsde
torchdiffeq==0.2.3
torchsde==0.2.6
transformers==4.30.2
pillow-avif-plugin==1.4.3
# 额外补充(在 versions 里有,但原 requirements.txt 里没有)
httpcore==0.15
spandrel==0.3.4
spandrel-extra-arches==0.1.1
httpx==0.24.1
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