Update Dockerfile

Dockerfile

@@ -1,10 +1,6 @@
 # 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
-
 # Set the working directory inside the container
 WORKDIR /app
 
@@ -12,19 +8,20 @@ WORKDIR /app
 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 \
+RUN 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 \
+    && wget -O /etc/apt/sources.list.d/nvidia-docker.list https://nvidia.github.io/nvidia-docker/$(. /etc/os-release; echo $ID$VERSION_ID)/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}"
+# Create a non-root user
+RUN useradd -m appuser
+
+# Grant necessary permissions to the non-root user
+RUN chown -R appuser /app /usr/local/
+
+USER appuser
 
 # Add uvicorn to the PATH
 ENV PATH="/home/appuser/.local/bin:${PATH}"