Update Dockerfile

Dockerfile

@@ -4,12 +4,17 @@ FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
 # Set Python to run in unbuffered mode for better logging
 ENV PYTHONUNBUFFERED=1
 
-# Install Python 3.10, pip, and system dependencies in a single layer
-RUN chmod 1777 /tmp \
-    && apt update -q && apt install -y ca-certificates wget libgl1 \
-    && wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
-    && dpkg -i /tmp/cuda-keyring.deb && apt update -q \
-    && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
+# Install Python and system dependencies in a single layer
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.10 \
+    python3-pip \
+    libgl1 \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    && ln -s /usr/bin/python3.10 /usr/bin/python \
+    && rm -rf /var/lib/apt/lists/*
 
 # Copy and install Python requirements
 COPY requirements.txt .