Update Dockerfile

Dockerfile

@@ -23,11 +23,14 @@ USER user
 COPY --chown=user public-apps/ /public-apps
 COPY --chown=user plates/ /home/user/plates
 
-RUN nvcc --version
-RUN apt-cache policy libcudnn8 | head -n 3
-
 USER root
-RUN apt-get update -y && apt-get upgrade -y && apt-get install -y libcudnn8
+RUN CUDNN_TAR_FILE="cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz" && \
+    wget wget https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz && \
+    tar -xzvf ${CUDNN_TAR_FILE}
+
+    sudo cp -P cuda/include/cudnn.h /usr/local/cuda-12.6/include && \
+    sudo cp -P cuda/lib/libcudnn* /usr/local/cuda-12.6/lib64/ && \
+    sudo chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
 USER user
 
 RUN /app/bin/warmup_apps