Update Dockerfile

Dockerfile

@@ -12,15 +12,15 @@ RUN apt-get update -q && \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# CUDA/CuDNN setup
+# CUDA/CuDNN setup with held packages override
 RUN 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-get update -q && \
-    apt-get install -y --no-install-recommends \
-    libcudnn8 \
-    libcublas-12-2 && \
-    rm -rf /var/lib/apt/lists/*
+    apt-get install -y --allow-change-held-packages --no-install-recommends \
+    libcudnn8=8.9.5.29-1+cuda12.2 \
+    libcublas-12-2=12.2.5.6-1 \
+    && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app