Update Dockerfile

Dockerfile

@@ -4,15 +4,12 @@
 FROM nvidia/cuda:11.7.1-runtime-ubuntu22.04
 
 RUN apt-get update && apt-get upgrade -y \
-    && apt-get install -y git build-essential libpq-dev \
-    python3 python3-pip gcc wget \
-    ocl-icd-opencl-dev opencl-headers clinfo \
-    libclblast-dev libopenblas-dev \
+    && apt-get install -y git build-essential libpq-dev gcc \
+    wget ocl-icd-opencl-dev opencl-headers clinfo \
+    libclblast-dev libopenblas-dev software-properties-common\
+    g++-11 make python3 python-is-python3 pip \
     && mkdir -p /etc/OpenCL/vendors && echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd
 
-RUN apt-get update && apt-get install -y software-properties-common
-RUN apt-get install -y g++-11
-
 ENV TRANSFORMERS_CACHE=/root/.cache
 
 COPY . .
@@ -22,7 +19,7 @@ ENV CUDA_DOCKER_ARCH=all
 ENV LLAMA_CUBLAS=1
 
 # Install depencencies
-RUN python3 -m pip install --upgrade pip pytest cmake \
+RUN python -m pip install --upgrade pip pytest cmake \
     scikit-build setuptools fastapi uvicorn sse-starlette \
     pydantic-settings starlette-context gradio huggingface_hub hf_transfer