Attempt to get working runpod CUDA12 dockerfile

nomadkaraoke · Jul 24, 2024 · 7e79526 · 7e79526
1 parent 3fc1118
commit 7e79526
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Dockerfile.runpod-cuda12 b/Dockerfile.runpod-cuda12
@@ -1,9 +1,15 @@
 # Runpod Base image: https://github.com/runpod/containers/blob/main/official-templates/base/Dockerfile
 FROM runpod/base:0.6.2-cuda12.1.0
 
-RUN apt-get update && apt-get install -y \
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
+RUN dpkg -i cuda-keyring_1.1-1_all.deb
+
+RUN apt-get update && apt-get upgrade -y
+
+RUN apt-get install -y \
     ffmpeg \
-    && rm -rf /var/lib/apt/lists/*
+    cuda-toolkit \
+    cudnn9-cuda-12
 
 RUN python3 -m pip install --upgrade pip