updateDKfile

Dockerfile

@@ -1,17 +1,38 @@
 FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
-
 RUN apt-get update && \
-    apt-get install -y python3-pip python3.8 && \
-    rm -rf /var/lib/apt/lists/*
-
-RUN python -m pip install --upgrade pip
+    apt-get upgrade -y && \
+    apt-get install -y --no-install-recommends \
 
-WORKDIR /code
-
-COPY ./requirements.txt /code/requirements.txt
+    # ffmpeg \
+    ffmpeg \
+  
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:${PATH}
+WORKDIR ${HOME}/app
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN curl https://pyenv.run | bash
+ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
+ENV PYTHON_VERSION=3.8.8
+RUN pyenv install ${PYTHON_VERSION} && \
+    pyenv global ${PYTHON_VERSION} && \
+    pyenv rehash && \
+    pip install --no-cache-dir -U pip setuptools wheel
 
-COPY . .
+RUN pip install --no-cache-dir -U torch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1
+COPY --chown=1000 requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
 
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+COPY --chown=1000 . ${HOME}/app
+RUN ls -a
+ENV PYTHONPATH=${HOME}/app \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
+CMD ["python", "app.py"]

requirements.txt

@@ -1,6 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.12.1+cu113 
-torchvision==0.13.1+cu113 
 torchaudio==0.12.1 
 torch-ema
 ninja