Use CUDA base image for better GPU compatibility

Dockerfile

@@ -1,12 +1,19 @@
-FROM python:3.11-slim
+FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
 
-WORKDIR /app
-
-# Install system dependencies
+# Install Python 3.11
 RUN apt-get update && apt-get install -y \
+    python3.11 \
+    python3.11-dev \
+    python3-pip \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
+# Create symlinks for python and pip
+RUN ln -s /usr/bin/python3.11 /usr/bin/python && \
+    ln -s /usr/bin/pip3 /usr/bin/pip
+
+WORKDIR /app
+
 # Copy requirements and install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt