update cuda and latest torch

.gitlab-ci.yml

@@ -38,10 +38,10 @@ build-xformers-8.6:
   variables: 
     GRAPHIC_VERSION: '8.6' # 30xx or A4000 A5000
 
-#build-xformers-8.9:
-#  extends: .build-xformers
-#  variables: 
-#    GRAPHIC_VERSION: '8.9' # 40xx
+build-xformers-8.9:
+  extends: .build-xformers
+  variables: 
+    GRAPHIC_VERSION: '8.9' # 40xx
 
 build-xformers-6.1:
   extends: .build-xformers
@@ -85,10 +85,10 @@ deploy-xformers-8.6:
   variables: 
     GRAPHIC_VERSION: '8.6'
 
-#deploy-xformers-8.9:
-#  extends: .deploy-xformers
-#  variables: 
-#    GRAPHIC_VERSION: '8.9'
+deploy-xformers-8.9:
+  extends: .deploy-xformers
+  variables: 
+    GRAPHIC_VERSION: '8.9'
 
 deploy-xformers-6.1:
   extends: .deploy-xformers

Dockerfile

-FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 as base
+FROM nvidia/cuda:12.0.1-cudnn8-devel-ubuntu22.04 as base
 
 RUN apt update \
   && apt-key adv --fetch-keys \
@@ -8,14 +8,14 @@ RUN apt update \
 RUN useradd -ms /bin/bash sd
 USER sd
 
-RUN pip install --no-cache-dir torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
+RUN pip install --no-cache-dir torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118
 
 RUN mkdir /home/sd/stable-diffusion-webui
 WORKDIR /home/sd/stable-diffusion-webui
 RUN mkdir repositories && \
 
   git clone --depth=1 https://github.com/Stability-AI/stablediffusion.git repositories/stable-diffusion-stability-ai && \
-  cd repositories/stable-diffusion-stability-ai && git fetch origin 47b6b607fdd31875c9279cd2f4f16b92e4ea958e && git checkout 47b6b607fdd31875c9279cd2f4f16b92e4ea958e && \
+  cd repositories/stable-diffusion-stability-ai && git fetch origin cf1d67a6fd5ea1aa600c4df58e5b47da45f6bdbf && git checkout cf1d67a6fd5ea1aa600c4df58e5b47da45f6bdbf && \
   python3 setup.py install --user && \
   cd ../.. && \
 
@@ -32,7 +32,7 @@ RUN mkdir repositories && \
   cd ../.. && \
 
   git clone --depth=1 https://github.com/crowsonkb/k-diffusion.git repositories/k-diffusion && \
-  cd repositories/k-diffusion && git fetch origin 5b3af030dd83e0297272d861c19477735d0317ec && git checkout 5b3af030dd83e0297272d861c19477735d0317ec && \
+  cd repositories/k-diffusion && git fetch origin c9fe758757e022f05ca5a53fa8fac28889e4f1cf && git checkout c9fe758757e022f05ca5a53fa8fac28889e4f1cf && \
   cd ../.. && \
 
   git clone --depth=1 https://github.com/Hafiidz/latent-diffusion repositories/latent-diffusion && \
@@ -59,7 +59,7 @@ RUN mkdir repositories && \
     git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b \
 
     # install xformers
-    # git+https://github.com/facebookresearch/xformers.git@v0.0.13#egg=xformers \
+    # git+https://github.com/facebookresearch/xformers.git@v0.0.20#egg=xformers \
 
     # install deepdanbooru
     git+https://github.com/KichangKim/DeepDanbooru.git@05eb3c39b0fae43e3caf39df801615fe79b27c2f#egg=deepdanbooru[tensorflow] tensorflow tensorflow-io \