Merge pull request #535 from transformerlab/fix/docker-base-image

deep1401 · web-flow · commit a7c4fc781536 · 2025-09-02T12:56:13.000-07:00
Fix base docker image
diff --git a/docker/common/Dockerfile b/docker/common/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.8.1-devel-ubuntu22.04
 
 SHELL ["/bin/bash", "--login", "-c"]
 
diff --git a/docker/gpu/nvidia/Dockerfile.cuda b/docker/gpu/nvidia/Dockerfile.cuda
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
+FROM nvidia/cuda:12.8.1-devel-ubuntu22.04
 
 SHELL ["/bin/bash", "--login", "-c"]
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM nvidia/cuda:12.1.1-devel-ubuntu22.04`
	`1`	`+FROM nvidia/cuda:12.8.1-devel-ubuntu22.04`
`2`	`2`
`3`	`3`	`SHELL ["/bin/bash", "--login", "-c"]`
`4`	`4`