Skip to content

Commit

Permalink
add k8s device plugin / GPU operator values file (#308)
Browse files Browse the repository at this point in the history
  • Loading branch information
samos123 authored Nov 10, 2024
1 parent c30396a commit 49f453e
Show file tree
Hide file tree
Showing 2 changed files with 31 additions and 0 deletions.
21 changes: 21 additions & 0 deletions charts/kubeai/values-nvidia-k8s-device-plugin.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
resourceProfiles:
nvidia-gpu-a16:
nodeSelector:
nvidia.com/gpu.family: "ampere"
nvidia.com/gpu.memory: "16384"
nvidia-gpu-l4:
nodeSelector:
nvidia.com/gpu.family: "ampere"
nvidia.com/gpu.memory: "23034"
nvidia-gpu-h100:
nodeSelector:
nvidia.com/gpu.family: "hopper"
nvidia.com/gpu.memory: "81920"
nvidia-gpu-a100-80gb:
nodeSelector:
nvidia.com/gpu.family: "ampere"
nvidia.com/gpu.memory: "81920"
nvidia-gpu-a100-40gb:
nodeSelector:
nvidia.com/gpu.family: "ampere"
nvidia.com/gpu.memory: "40960"
10 changes: 10 additions & 0 deletions charts/kubeai/values.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -137,6 +137,16 @@ resourceProfiles:
operator: "Equal"
value: "present"
effect: "NoSchedule"
nvidia-gpu-a16:
imageName: "nvidia-gpu"
limits:
nvidia.com/gpu: "1"
tolerations:
- key: "nvidia.com/gpu"
operator: "Equal"
value: "present"
effect: "NoSchedule"


cacheProfiles: {}

Expand Down

0 comments on commit 49f453e

Please sign in to comment.