diff --git a/Dockerfile b/Dockerfile
index 20f1cfa..33eabac 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,6 @@
 
-FROM nvidia/cuda:11.8.0-devel-ubuntu20.04
+# FROM nvidia/cuda:11.8.0-devel-ubuntu20.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
 
 LABEL maintainer Ilija Vukotic <ivukotic@cern.ch>
 
@@ -34,11 +35,11 @@ ARG OS=ubuntu2004
 ARG cudnn_version=8.6.0
 ARG cuda_version=cuda11.8
 
-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 && apt-get install libcudnn8=${cudnn_version}-1+${cuda_version}
+# 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 && apt-get install libcudnn8=${cudnn_version}-1+${cuda_version}
 
 
 # make sure we have a way to bind host provided libraries