Update Dockerfile

Dockerfile

@@ -1,7 +1,22 @@
-FROM python:3.10.12
+FROM nvidia/cuda:11.8.0-base-ubuntu22.04
 
 WORKDIR /code
 
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt update 
+
+RUN apt install software-properties-common -y 
+
+RUN add-apt-repository ppa:deadsnakes/ppa 
+
+RUN apt install -y python3.9 python3.9-distutils python3-pip
+
+# Install pip 23.1.2
+RUN python3.9 -m pip install pip==23.1.2
+
+RUN apt-get install -y git-all
+
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
@@ -17,4 +32,4 @@ WORKDIR $HOME/app
 
 COPY --chown=user . $HOME/app
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]