Update Dockerfile

Dockerfile

@@ -1,30 +1,36 @@
-FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:10.1-cudnn7-runtime
 
-ARG DEBIAN_FRONTEND=noninteractive
-
-RUN apt update
-RUN apt install -y git python python3 python3-pip wget
-
-EXPOSE 7860
-
-RUN export PATH="/usr/local/cuda/bin:$PATH"
-RUN wget -qO- "https://cmake.org/files/v3.17/cmake-3.17.0-Linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C /usr/local
-
-RUN useradd -m -u 1000 user
-
-USER user
+# Set the working directory
+WORKDIR $HOME/app
 
-ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH
+# Copy the requirements file
+COPY requirements.txt .
 
-WORKDIR $HOME/app
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        build-essential \
+        python3-dev \
+        python3-pip \
+        libglib2.0-0 && \
+    rm -rf /var/lib/apt/lists/*
 
 RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install llama-cpp-python --no-cache-dir
-COPY requirements.txt ./
 
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir --upgrade -r requirements.txt
 
+# Copy the installed packages to a new image
+FROM nvidia/cuda:10.1-cudnn7-runtime
+COPY --from=0 /root/.local /root/.local
+ENV PATH=/root/.local/bin:$PATH
+
+EXPOSE 7860
+
+RUN useradd -m -u 1000 user
+USER user
+
+# Set the working directory and copy the application code
+WORKDIR /app
 COPY --chown=user . $HOME/app
 
 RUN ls -al