Update Dockerfile

Dockerfile

@@ -1,5 +1,21 @@
 #FROM python:3.11-slim-bullseye
-FROM huggingface/optimum-nvidia
+FROM nvidia/cuda:12.3.0-devel-ubuntu22.04
+
+RUN apt-get update && \
+    apt-get -y install \
+      git \
+    python3.10 \
+    python3-pip \
+    openmpi-bin \
+    libopenmpi-dev
+
+COPY . /opt/optimum-nvidia
+WORKDIR /opt/optimum-nvidia
+
+RUN pip install --pre --extra-index-url https://pypi.nvidia.com -e '.[quality, tests]' && \
+    pip uninstall -y optimum-nvidia && \
+    rm -rf /opt/optimum-nvidia
+    
 RUN apt-get update && apt-get upgrade -y && apt-get install -y --no-install-recommends \
    python3 \
     python3-pip \