Update Dockerfile

Dockerfile

@@ -1,8 +1,17 @@
-FROM nvidia/cuda:11.7.0-cudnn8-runtime-ubuntu22.04
-
-# Install Python
+FROM python:3.10-slim
+
+# 1) Install CUDA runtime 11.7
+RUN apt-get update && apt-get install -y gnupg curl && \
+    curl -s -L https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb \
+      -o cuda-keyring.deb && \
+    dpkg -i cuda-keyring.deb && \
+    apt-get update && \
+    apt-get install -y cuda-runtime-11-7 && \
+    rm -rf /var/lib/apt/lists/* cuda-keyring.deb
+
+# install build/c runtime deps
 RUN apt-get update && apt-get install -y \
-      python3 python3-pip python3-dev \
+      build-essential python3-dev git \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
@@ -19,7 +28,7 @@ ENV \
 RUN mkdir -p /app/.cache/{pip,torch,numba} \
  && chmod -R 777 /app/.cache
 
-# 1) install ALL your Python deps (including CPU-only torch)
+# 2) install ALL your Python deps (including CPU-only torch)
 COPY requirements.txt .
 RUN pip install --upgrade pip \
  && pip install --no-cache-dir -r requirements.txt