Use NVIDIA CUDA base image for GPU support

Dockerfile

@@ -1,16 +1,20 @@
-FROM python:3.11-slim
+FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
 
 WORKDIR /app
 
-# Fix OMP environment variable issue
+# Fix environment variables
 ENV OMP_NUM_THREADS=4
 ENV MKL_NUM_THREADS=4
+ENV DEBIAN_FRONTEND=noninteractive
 
-# Install system dependencies
+# Install Python and dependencies
 RUN apt-get update && apt-get install -y \
-    build-essential \
+    python3.11 \
+    python3-pip \
+    python3.11-venv \
     git \
-    && rm -rf /var/lib/apt/lists/*
+    && rm -rf /var/lib/apt/lists/* \
+    && ln -sf /usr/bin/python3.11 /usr/bin/python
 
 # Copy requirements and install
 COPY requirements.txt .