Update Dockerfile

Dockerfile

@@ -1,23 +1,49 @@
-FROM nvidia/cuda:12.2.0-base-ubuntu22.04
-ENV DEBIAN_FRONTEND noninteractive
+FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
 
-RUN apt-get update -y && \
-	apt-get install -y aria2 libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 && \
-	pip install -q torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 torchtext==0.15.2 torchdata==0.6.1 --extra-index-url https://download.pytorch.org/whl/cu118 && \
-	pip install xformers==0.0.20 triton==2.0.0 packaging==23.1 && \
-	adduser --disabled-password --gecos '' user && \
-	mkdir /content && \
-	chown -R user:user /content
+ENV DEBIAN_FRONTEND=noninteractive \
+    TZ=America/Los_Angeles
 
+ARG USE_PERSISTENT_DATA
+
+RUN apt-get update && apt-get install -y \
+    git libgl1 libglib2.0-0 \
+    make build-essential libssl-dev zlib1g-dev \
+    libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
+    libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git git-lfs  \
+    ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
+    && rm -rf /var/lib/apt/lists/* \
+    && git lfs install \
+    apt-get install nvidia-container-runtime
+    
 WORKDIR /content
+
+# User
+RUN useradd -m -u 1000 user
 USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+# Pyenv
+RUN curl https://pyenv.run | bash
+ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
 
-RUN git clone -b v2.6 https://github.com/camenduru/stable-diffusion-webui /content/test && \
+ARG PYTHON_VERSION=3.10.12
+# Python
+RUN pyenv install $PYTHON_VERSION && \
+    pyenv global $PYTHON_VERSION && \
+    pyenv rehash && \
+    pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir \
+    datasets \
+    huggingface-hub "protobuf<4" "click<8.1"
+
+RUN git clone -b v3.0 https://github.com/camenduru/stable-diffusion-webui /content/test && \
 	git clone https://github.com/etherealxx/batchlinks-webui /content/test/extensions/batchlinks-webui  && \
 	sed -i -e 's/    start()/    #start()/g' /content/test/launch.py && \
 	cd /content/test && \
     python launch.py --skip-torch-cuda-test && \
     git reset --hard
 
-# RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd15/resolve/main/v1-5-pruned-emaonly.ckpt -d /content/test/models/Stable-diffusion -o v1-5-pruned-emaonly.ckpt
+RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd15/resolve/main/v1-5-pruned-emaonly.ckpt -d /content/test/models/Stable-diffusion -o v1-5-pruned-emaonly.ckpt
+
 CMD cd /content/test && python launch.py --xformers --cors-allow-origins=* --api