Update Dockerfile

Dockerfile

@@ -1,9 +1,6 @@
 # Use an official Python runtime as a parent image
 FROM python:3.9-slim
 
-# Use an official Python runtime as a parent image
-FROM python:3.9-slim
-
 # Create a non-root user
 RUN useradd -m appuser
 USER appuser
@@ -17,6 +14,18 @@ COPY ./requirements.txt /app/
 # Install the dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Add NVIDIA Docker repository and install NVIDIA Docker
+RUN distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
+    && apt-get update && apt-get install -y wget gnupg \
+    && wget -O /tmp/NVIDIA-DOCKER-KEY https://nvidia.github.io/nvidia-docker/gpgkey \
+    && apt-key add /tmp/NVIDIA-DOCKER-KEY \
+    && rm -f /tmp/NVIDIA-DOCKER-KEY \
+    && wget -O /etc/apt/sources.list.d/nvidia-docker.list https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list \
+    && apt-get update && apt-get install -y nvidia-docker2
+
+# Add NVIDIA Docker to the PATH
+ENV PATH="/usr/bin/nvidia-docker:${PATH}"
+
 # Add uvicorn to the PATH
 ENV PATH="/home/appuser/.local/bin:${PATH}"
 
@@ -26,5 +35,5 @@ COPY . /app/
 # Create a writable cache directory inside the /app directory
 RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
 
-# Specify the command to run your application using NVIDIA Container Toolkit and uvicorn
-CMD ["nvidia-docker", "run", "--rm", "-p", "8000:8000", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+# Specify the command to run your application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]