Skip to content

Commit

Permalink
add default configuration to allow scheduling on gpu nodes in GKE
Browse files Browse the repository at this point in the history
Signed-off-by: Cornelis Boon <[email protected]>
  • Loading branch information
cjidboon94 committed Mar 21, 2024
1 parent aeb4d55 commit 8a6d202
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions charts/flyte-binary/gke-starter.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,13 @@ configuration:
plugins:
k8s:
inject-finalizer: true
gpu-device-node-label: cloud.google.com/gke-accelerator
resource-tolerations:
- nvidia.com/gpu:
- key: "nvidia.com/gpu"
operator: "Equal"
value: "present"
effect: "NoSchedule"
# Configuration for the Datacatalog engine, used when caching is enabled
# Learn more: https://docs.flyte.org/en/latest/deployment/configuration/generated/datacatalog_config.html
storage:
Expand Down

0 comments on commit 8a6d202

Please sign in to comment.