Updated docker file to be multi-stage build

.dockerignore

@@ -1,2 +1,3 @@
 /venv
 /flagged
+/.env

Dockerfile

@@ -1,18 +1,30 @@
-FROM continuumio/miniconda3
-EXPOSE 7860
+FROM python:3.7-slim as builder
+
+RUN apt update && \
+    apt install --no-install-recommends -y build-essential gcc
+COPY requirements_personal.txt /req.txt
 
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends libgl1-mesa-glx && \
-    apt-get install -y --no-install-recommends libglib2.0-0 && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
+RUN python -m pip install --upgrade pip && \
+    pip install --no-cache-dir --user -r req.txt && \
+    pip install --no-cache-dir --user typing-extensions -U
 
-ENV GRADIO_SERVER_NAME=0.0.0.0
-WORKDIR /workspace
+# Stage 2: Runtime
+FROM nvidia/cuda:11.3.1-cudnn8-runtime
 
-ADD environment.yaml /workspace/environment.yaml
-RUN conda env update -n base --file environment.yaml && \
-    pip install typing-extensions -U
+RUN rm /etc/apt/sources.list.d/cuda.list
+RUN rm /etc/apt/sources.list.d/nvidia-ml.list
 
-ADD app.py /workspace/
-CMD [ "python" , "/workspace/app.py" ]
+ENV TZ=America/New_York
+RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
+
+RUN apt update && \
+    apt install --no-install-recommends -y build-essential software-properties-common && \
+    add-apt-repository -y ppa:deadsnakes/ppa && \
+    apt install --no-install-recommends -y python3.7 python3-distutils && \
+    # update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 1 && \
+    # update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.7 2 && \
+    apt clean && rm -rf /var/lib/apt/lists/*
+COPY --from=builder /root/.local/lib/python3.7/site-packages /usr/local/lib/python3.7/dist-packages
+COPY app.py app.py
+CMD ["python3", "app.py"]
+EXPOSE 7860

Dockerfile.bkup

@@ -0,0 +1,18 @@
+FROM continuumio/miniconda3
+EXPOSE 7860
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libgl1-mesa-glx && \
+    apt-get install -y --no-install-recommends libglib2.0-0 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+ENV GRADIO_SERVER_NAME=0.0.0.0
+WORKDIR /workspace
+
+ADD environment.yaml /workspace/environment.yaml
+RUN conda env update -n base --file environment.yaml && \
+    pip install typing-extensions -U
+
+ADD app.py /workspace/
+CMD [ "python" , "/workspace/app.py" ]