From 4603bb64a7cc037fbbca547687352ce307182126 Mon Sep 17 00:00:00 2001
From: Malte Grosse <malte.grosse@googlemail.com>
Date: Tue, 28 Nov 2023 22:48:30 +0900
Subject: [PATCH] 4

---
 .woodpecker.yml | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/.woodpecker.yml b/.woodpecker.yml
index 80e45bb..27295ce 100644
--- a/.woodpecker.yml
+++ b/.woodpecker.yml
@@ -12,6 +12,12 @@ steps:
      - nvidia-smi
 #   environment:
 #     - NVIDIA_VISIBLE_DEVICES=all  
+    backend_options:
+      kubernetes:
+        resources:
+          limits:
+            nvidia.com/mig-2g.20gb: 1
+
   gpu:
     image: oguzpastirmaci/gpu-burn:latest
  #   environment:
@@ -20,7 +26,11 @@ steps:
       - cd /app
       - ./gpu_burn 120
       - echo "burned. done"
-
+    backend_options:
+      kubernetes:
+        resources:
+          limits:
+            nvidia.com/mig-2g.20gb: 1
 #  2nd:
 #    image: oguzpastirmaci/gpu-burn:latest
 #    environment: