Update Dockerfile

Dockerfile

@@ -1,10 +1,7 @@
-FROM python:3.9
+FROM nvidia/cuda:12.1.1-devel-ubi8
 WORKDIR /app
 
-RUN distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
-  && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add - \
-  && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list \
-  && apt-get update && apt-get install -y nvidia-container-toolkit
+RUN apt update && apt install python3 python3-pip
 
 COPY ./requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt