Update Dockerfile

Dockerfile

@@ -1,18 +1,19 @@
 # Use an NVIDIA CUDA image as the base
 FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
-# Install Python and pip
+# Set the working directory in the container
+WORKDIR /code
+
+# Install Python, pip, and all required build tools in a single step
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.10 \
     python3-pip \
+    gcc \
+    g++ \
+    make \
+    cmake \
     && rm -rf /var/lib/apt/lists/*
 
-# Set the working directory in the container
-WORKDIR /code
-
-# Install build tools needed for llama-cpp-python
-RUN apt-get update && apt-get install -y --no-install-recommends gcc g++ make
-
 # Set environment variables for GPU-accelerated build
 ENV CMAKE_ARGS="-DGGML_CUDA=on"
 ENV FORCE_CMAKE=1