try prebuilt cudnn

ivukotic · May 22, 2023 · 935644a · 935644a
1 parent 96900e9
commit 935644a
Showing 1 changed file with 7 additions and 6 deletions.
diff --git 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 <[email protected]>
 
@@ -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