Update Dockerfile

Dockerfile

@@ -1,9 +1,10 @@
 FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 AS llama_builder
 
+ARG DEBIAN_FRONTEND=noninteractive
+
 RUN apt-get update && apt-get install --no-install-recommends -y \
   build-essential \
   python3-pip \
-  python3-dev \
   && apt-get clean && rm -rf /var/lib/apt/lists/*
 
 RUN python -m venv /opt/venv
@@ -15,6 +16,10 @@ RUN CMAKE_ARGS="-DLLAMA_CUDA=on" pip3 install --no-cache-dir llama-cpp-python
 
 FROM nvcr.io/nvidia/pytorch:23.10-py3 AS dep_builder
 
+RUN apt-get update && apt-get install --no-install-recommends -y \
+  python3-pip \
+  && apt-get clean && rm -rf /var/lib/apt/lists/*
+
 RUN python -m venv /opt/venv
 ENV PATH=/opt/venv/bin:$PATH \
     NVIDIA_VISIBLE_DEVICES=all \
@@ -30,6 +35,7 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
   ffmpeg \
   libsm6 \
   libxext6 \
+  python3.10
   python3-pip \
   python3-dev \
   ninja-build \