Update Dockerfile

Dockerfile

@@ -1,21 +1,28 @@
-FROM python:3.10.9
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04
 
-# Use a more secure method for setting the working directory
+# Install Python and other dependencies
+RUN apt-get update && apt-get install -y \
+    python3.10 \
+    python3-pip \
+    ffmpeg \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set up the working directory
 WORKDIR /app
 
-# Copy the requirements file
+# Copy the requirements file and install dependencies
 COPY ./requirements.txt /app/requirements.txt
-
-# Install dependencies
-RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
-RUN apt update && apt install -y ffmpeg
+RUN python3 -m pip install --upgrade pip \
+    && python3 -m pip install --no-cache-dir --upgrade -r /app/requirements.txt
 
 # Add and configure the user
 RUN useradd -m -u 1000 user
 
 # Set environment variables for the user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
+    PATH=/home/user/.local/bin:$PATH \
+    CUDA_HOME=/usr/local/cuda
 
 # Set the user and working directory
 USER user
@@ -30,7 +37,6 @@ RUN chmod -R 777 $HOME/app
 # Set additional environment variables
 ENV NUMBA_CACHE_DIR=/tmp/NUMBA_CACHE_DIR/
 ENV MPLCONFIGDIR=/tmp/MPLCONFIGDIR/
-ENV HOME=/home/user
 ENV PATH="/opt/venv/bin:$PATH"
 
 # Set the command to run the application