Update Dockerfile

Dockerfile

@@ -2,15 +2,11 @@ FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-RUN apt-get update -q && \
-    apt-get install -y --no-install-recommends \
-    python3-pip \
-    python3-dev \
-    ffmpeg \
-    libgl1 \
-    git \
-    wget \
-    && rm -rf /var/lib/apt/lists/*
+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
 
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt