Update docker-bake.hcl

runpod · Nov 1, 2023 · 5965cb4 · 5965cb4
1 parent 70cdd86
commit 5965cb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/official-templates/base/docker-bake.hcl b/official-templates/base/docker-bake.hcl
@@ -24,7 +24,7 @@ target "11-8-0" {
         logo = "../../container-template"
     }
     args = {
-        BASE_IMAGE = "nvidia/11.8.0-devel-ubuntu20.04"
+        BASE_IMAGE = "nvidia/cuda:11.8.0-devel-ubuntu20.04"
     }
 }
 
@@ -37,6 +37,6 @@ target "12-1-0" {
         logo = "../../container-template"
     }
     args = {
-        BASE_IMAGE = "nvidia/12.1.0-devel-ubuntu22.04"
+        BASE_IMAGE = "nvidia/cuda:12.1.0-devel-ubuntu22.04"
     }
 }