update: pipeline run

redhat-na-ssa · Oct 3, 2024 · 87ee935 · 87ee935
1 parent d4dfe20
commit 87ee935
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/scripts/pipeline-run.yaml b/scripts/pipeline-run.yaml
@@ -7,6 +7,14 @@ metadata:
   labels:
     tekton.dev/pipeline: train-model
 spec:
+  podTemplate:
+    tolerations:
+      - key: nvidia-gpu-only
+        operator: Exists
+        effect: NoSchedule
+      - key: nvidia.com/gpu
+        operator: Exists
+        effect: NoSchedule
   params:
     - name: BATCH_SIZE
       value: '-1'