Base image support for CUDA 12.1 / Python 3.11 / Torch 2.1.2

Signed-off-by: andreasjansson <[email protected]>
andreasjansson · Apr 12, 2024 · 72398c7 · 72398c7
1 parent 488e117
commit 72398c7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/dockerfile/base.go b/pkg/dockerfile/base.go
@@ -162,6 +162,7 @@ func BaseImageConfigurations() []BaseImageConfiguration {
 		{"12.1", "3.11", ""},
 		{"12.1", "3.11", "2.1.0"},
 		{"12.1", "3.11", "2.1.1"},
+		{"12.1", "3.11", "2.1.2"},
 		{"12.1", "3.9", ""},
 		{"12.1", "3.9", "2.1.0"},
 		{"12.1.1", "3.11", "2.1.1"},