Use pushed base-gpu (#189)

Author: JonasHedEng

src/container/BUILD.gpu

@@ -10,6 +10,14 @@ oci_pull_images(
     variants=variants,
 )
 
+oci_pull_images(
+    name="base-gpu-repo",
+    repository="europe-docker.pkg.dev/embark-shared/main/ml/base-gpu",
+    variants={
+        "latest": "909f971eb81cbd145f64e8db59d55fee4405a6ce6404dd61c9a049fbcc2eb88d",
+    },
+)
+
 # Useful for debugging builds
 # oci_pull_images(
 #     name="ubuntu",
@@ -40,6 +48,8 @@ oci_image_build(
                 -newermt "@${SOURCE_DATE_EPOCH}" -writable -xdev \
             | xargs touch --date="@${SOURCE_DATE_EPOCH}" --no-dereference \
     """,
+    repository="europe-docker.pkg.dev/embark-shared/main/ml/base-gpu",
+    tag="latest",
     entrypoint='["python3.10", "/bin/app/__main__.py"]',
 )
 
@@ -108,7 +118,7 @@ oci_build_layer(
 
 oci_image_build(
     name="erupt-runner-gpu",
-    base=[":base-gpu"],
+    base=[":base-gpu-repo#latest"],
     commands="""
         mkdir -p /bin/app/bin && ln -sf /usr/bin/python3.10 /bin/app/bin/python3.10 \
     """,