Autopilot Clusters with GKE managed DRANET: GPUs and TPUs

https://storage.googleapis.com/gweb-cloudblog-publish/images/0-hero_pfrvm6j.max-2500x2500.png

Google Kubernetes Engine (GKE) managed DRANET supports both GPUs and TPUs. There are several configurations to use this implementation, including standard cluster (where you have full control) and autopilot cluster (where Google does the heavy configs for you). I've been exploring the capabilities and in this blog we will explore setting up for autopilot clusters.

Autopilot and managed DRANET

GKE autopilot is a managed version of GKE that handles nodes, scaling, security, and other preconfigured settings. GKE managed DRANET lets you request and allocate networking resources for your Pods, including network interfaces that support TPUs and Remote Direct Memory Access (RDMA).

Setup flow

To deploy your GKE autopilot cluster and enable managed DRANET, you need to create a Virtual Private Cloud (VPC). Let's walk through the setup:

  1. Deploy an Autopilot cluster.
  2. Create a custom ComputeClass which supports the accelerator type (TPU or GPU)
  3. Create a ResourceClaimTemplatefor GPUs (RDMA)...

Copyright of this story solely belongs to google.com. To see the full text click HERE

Read more