From 935644adf67d688eecfd746e81603e7fbc05a76b Mon Sep 17 00:00:00 2001
From: Ilija Vukotic <ivukotic@uchicago.edu>
Date: Mon, 22 May 2023 01:04:51 -0500
Subject: [PATCH] try prebuilt cudnn

---
 Dockerfile | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

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