diff --git a/.docker/Dockerfile b/.docker/Dockerfile
index 6759d8ae8..628583766 100644
--- a/.docker/Dockerfile
+++ b/.docker/Dockerfile
@@ -4,19 +4,21 @@ ENV OS=ubuntu2004
 ENV cudnn_version=8.9.4.25
 ENV cuda_version=cuda12.2
 
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y wget
+
 RUN wget https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin
 
 RUN mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/7fa2af80.pub
 RUN add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/ /"
-RUN apt-get update
+
 
 RUN apt-get update &&  \
-    DEBIAN_FRONTEND=noninteractive apt-get install -y openjdk-11-jdk wget \
-    build-essential checkinstall zlib1g-dev libssl-dev git libpthread-stubs0-dev
+    DEBIAN_FRONTEND=noninteractive apt-get install -y openjdk-11-jdk \
+    build-essential checkinstall zlib1g-dev libssl-dev git libpthread-stubs0-dev \
+    libcudnn8=${cudnn_version}-1+${cuda_version} libcudnn8-dev=${cudnn_version}-1+${cuda_version}
 
 
-RUN apt-get install libcudnn8=${cudnn_version}-1+${cuda_version} libcudnn8-dev=${cudnn_version}-1+${cuda_version}
 #RUN apt-get install libcudnn8-samples=${cudnn_version}-1+${cuda_version}
     #Build cmake version from source \
 #RUN wget https://github.com/Kitware/CMake/releases/download/v3.24.2/cmake-3.24.2.tar.gz && \