Update docker-bake.hcl

runpod · Nov 3, 2023 · cd0319d · cd0319d
1 parent d2fac51
commit cd0319d
Showing 1 changed file with 20 additions and 12 deletions.
diff --git a/official-templates/pytorch/docker-bake.hcl b/official-templates/pytorch/docker-bake.hcl
@@ -1,11 +1,3 @@
-variable "common_context" {
-  default = {
-    scripts = "../../container-template"
-    proxy = "../../container-template/proxy"
-    logo = "../../container-template"
-  }
-}
-
 group "default" {
     targets = [
         "1131-py38-cuda1171-devel",
@@ -19,7 +11,11 @@ group "default" {
 target "1131-py38-cuda1171-devel" {
     dockerfile = "Dockerfile"
     tags = ["runpod/pytorch:1.13.0-py3.10-cuda11.7.1-devel"]
-    contexts = ${common_context}
+    contexts = {
+        scripts = "../../container-template"
+        proxy = "../../container-template/proxy"
+        logo = "../../container-template"
+    }
     args = {
         BASE_IMAGE = "nvidia/cuda:11.8.0-devel-ubuntu22.04"
         PYTHON_VERSION = "3.8"
@@ -31,7 +27,11 @@ target "1131-py38-cuda1171-devel" {
 target "201-py310-cuda1180-devel" {
     dockerfile = "Dockerfile"
     tags = ["runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel"]
-    contexts = ${common_context}
+    contexts = {
+        scripts = "../../container-template"
+        proxy = "../../container-template/proxy"
+        logo = "../../container-template"
+    }
     args = {
         BASE_IMAGE = "nvidia/cuda:11.8.0-devel-ubuntu22.04"
         PYTHON_VERSION = "3.10"
@@ -43,7 +43,11 @@ target "201-py310-cuda1180-devel" {
 target "191-py39-cuda111-devel" {
     dockerfile = "Dockerfile"
     tags = ["runpod/pytorch:1.9.1-py3.9-cuda11.1.1-devel"]
-    contexts = ${common_context}
+    contexts = {
+        scripts = "../../container-template"
+        proxy = "../../container-template/proxy"
+        logo = "../../container-template"
+    }
     args = {
         BASE_IMAGE = "nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04"
         PYTHON_VERSION = "3.9"
@@ -55,7 +59,11 @@ target "191-py39-cuda111-devel" {
 target "210-py310-cuda1180-devel" {
     dockerfile = "Dockerfile"
     tags = ["runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel"]
-    contexts = ${common_context}
+    contexts = {
+        scripts = "../../container-template"
+        proxy = "../../container-template/proxy"
+        logo = "../../container-template"
+    }
     args = {
         BASE_IMAGE = "nvidia/cuda:11.8.0-devel-ubuntu20.04"
         PYTHON_VERSION = "3.10"