Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-# Use an official Python runtime as a parent image
-FROM python:3.9-slim
+# Use NVIDIA CUDA image as the parent image
+FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
 
 # Set the working directory inside the container
 WORKDIR /app
@@ -11,7 +11,7 @@ RUN useradd -m appuser
 COPY ./requirements.txt /app/
 
 # Grant necessary permissions to the non-root user
-RUN chown -R appuser /app /usr/local/
+RUN chown -R appuser /app
 
 # Switch to the non-root user
 USER appuser