Update nvidia/cuda Docker tag to v12

defenseunicorns · Jun 6, 2023 · 5a6e289 · 5a6e289
1 parent d4e23c2
commit 5a6e289
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/embeddings/Dockerfile b/embeddings/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
 # Update, install
 RUN apt update && \

diff --git a/llms/Dockerfile.template b/llms/Dockerfile.template
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
 # Update, install
 RUN apt update && \

diff --git a/llms/stablelm-7b/Dockerfile b/llms/stablelm-7b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
 # Update, install
 RUN apt update && \

diff --git a/llms/stablelm/Dockerfile b/llms/stablelm/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
 # Update, install
 RUN apt update && \

diff --git a/models/whisper/Dockerfile b/models/whisper/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
 # Update, install
 RUN apt update && \