add nvidia-gpu-8gb resource profile

substratusai · Dec 2, 2024 · 94fb6b4 · 94fb6b4
1 parent d834b1c
commit 94fb6b4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/charts/kubeai/values-nvidia-k8s-device-plugin.yaml b/charts/kubeai/values-nvidia-k8s-device-plugin.yaml
@@ -19,3 +19,6 @@ resourceProfiles:
     nodeSelector:
       nvidia.com/gpu.family: "ampere"
       nvidia.com/gpu.memory: "40960"
+  nvidia-gpu-8gb:
+    nodeSelector:
+      nvidia.com/gpu.memory: "8188"
diff --git a/charts/kubeai/values.yaml b/charts/kubeai/values.yaml
@@ -168,7 +168,15 @@ resourceProfiles:
         operator: "Equal"
         value: "present"
         effect: "NoSchedule"
-
+  nvidia-gpu-8gb:
+    imageName: "nvidia-gpu"
+    limits:
+      nvidia.com/gpu: "1"
+    tolerations:
+      - key: "nvidia.com/gpu"
+        operator: "Equal"
+        value: "present"
+        effect: "NoSchedule"
 
 cacheProfiles: {}