Crea un cluster GKE ottimizzato per l'AI che utilizza G2 o G4

Questo documento descrive come creare cluster Google Kubernetes Engine (GKE) ottimizzati per l'AI che utilizzano le serie di macchine ottimizzate per gli acceleratori G2 (NVIDIA L4) o G4 (NVIDIA RTX PRO 6000) per supportare i workload di intelligenza artificiale (AI) e machine learning (ML). G2 e G4 sono serie di macchine GPU generiche, che forniscono risorse di calcolo indipendenti progettate per le attività di AI mainstream, come l'inferenza di piccole e medie dimensioni e le applicazioni con un utilizzo elevato di grafica. Oltre ai tipi di macchine con una o più GPU, la serie di macchine G4 supporta le GPU virtuali (vGPU) sui tipi di macchine con meno di una GPU:

  • g4-standard-6 (un ottavo di GPU)
  • g4-standard-12 (un quarto di GPU)
  • g4-standard-24 (metà di una GPU)

GKE fornisce una singola piattaforma per eseguire un insieme diversificato di workload per la tua organizzazione, riducendo il carico operativo della gestione di più piattaforme.

Per creare un cluster GKE ottimizzato per l'AI con G2 o G4, completa i seguenti passaggi:

  1. Crea l'ambiente GKE
  2. Connettiti al tuo cluster
  3. Configura i manifest dei pod

Prima di iniziare

Prima di iniziare, assicurati di aver eseguito le seguenti operazioni:

  • Abilita l'API Google Kubernetes Engine.
  • Abilita l'API Google Kubernetes Engine
  • Per utilizzare Google Cloud CLI per questa attività, installala e poi inizializza gcloud CLI. Se hai già installato gcloud CLI, scarica l'ultima versione eseguendo il comando gcloud components update. Le versioni precedenti di gcloud CLI potrebbero non supportare l'esecuzione dei comandi in questo documento.

Ruoli obbligatori

Per ottenere le autorizzazioni necessarie per creare e gestire un cluster GKE, chiedi all'amministratore di concederti i seguenti ruoli IAM nel progetto:

Per saperne di più sulla concessione dei ruoli, consulta Gestisci l'accesso a progetti, cartelle e organizzazioni.

Potresti anche riuscire a ottenere le autorizzazioni richieste tramite i ruoli personalizzati o altri ruoli predefiniti.

Scegli un'opzione di consumo e ottieni la capacità

  1. Scegli un'opzione di consumo. Fai la tua scelta in base a come vuoi ottenere e utilizzare le risorse GPU. Per ulteriori informazioni, consulta la sezione Scegliere un'opzione di consumo.

    Per GKE, tieni presente le seguenti informazioni aggiuntive quando scegli un'opzione di consumo:

  2. Ottenere capacità. Scopri come ottenere capacità per l'opzione di consumo.

Requisiti

Per creare un cluster GKE ottimizzato per l'AI che utilizza G2 o G4, assicurati di soddisfare i seguenti requisiti:

  • Versione GKE:le macchine G4 (NVIDIA RTX PRO 6000) richiedono le seguenti versioni GKE minime:
    • Modalità Standard:
      • Tipi di macchine con una o più GPU: 1.34.0-gke.1662000 o versioni successive
      • Tipi di macchine con meno di una GPU (g4-standard-6, g4-standard-12 e g4-standard-24):
        • 1.35:1.35.8-gke.1518000 o versioni successive
        • 1.36 o versioni successive: 1.36.4-gke.1082000 o versioni successive
    • Modalità Autopilot:
      • Tipi di macchine con una o più GPU: 1.34.1-gke.1829001 o versioni successive
      • Tipi di macchine con meno di una GPU (g4-standard-6, g4-standard-12 e g4-standard-24):
        • 1.35:1.35.8-gke.1518000 o versioni successive
        • 1.36 o versioni successive: 1.36.4-gke.1082000 o versioni successive
  • Driver della GPU:
    • I nodi G2 (NVIDIA L4) devono utilizzare la versione del driver NVIDIA 535 o successive.
    • I nodi G4 (NVIDIA RTX PRO 6000) devono utilizzare la versione 580 o successive del driver NVIDIA.

Limitazioni

Le seguenti limitazioni si applicano alle serie di macchine G2 e G4 come GPU generiche:

  • SSD locali:i tipi di macchine G2 e G4 non includono dischi SSD locali per impostazione predefinita. Se necessario, devi allegarli manualmente. Il tipo di macchina g4-standard-6 (un ottavo di una GPU) non supporta le SSD locali.
  • NCCL Fast Socket:non puoi utilizzare NCCL Fast Socket con macchine G2 o G4 su GKE. Sebbene le macchine G2 e G4 supportino la comunicazione multi-nodo, utilizzano il networking VPC standard. Per l'addestramento distribuito su larga scala che richiede il massimo throughput di rete, ti consigliamo di utilizzare la serie di macchine GPU in cluster, come A3 High o A3 Mega (che utilizzano GPUDirect TCPX) o A3 Ultra e A4 (che utilizzano GPUDirect RDMA).
  • Tipi di macchine G4 con meno di una GPU:per g4-standard-6, g4-standard-12 e g4-standard-24:

Crea l'ambiente GKE

Puoi creare un cluster in modalità Autopilot o Standard.

Autopilot

Per creare un cluster Autopilot, esegui questo comando:

gcloud container clusters create-auto CLUSTER_NAME \
    --region=REGION

Sostituisci quanto segue:

  • CLUSTER_NAME: il nome del tuo cluster.
  • REGION: la regione del cluster.

Standard

Crea un cluster Standard e un pool di nodi GPU:

  1. Per creare un cluster Standard, esegui questo comando:

    gcloud container clusters create CLUSTER_NAME \
        --region=REGION \
        --enable-ip-alias
    

    Sostituisci quanto segue:

    • CLUSTER_NAME: il nome del tuo cluster.
    • REGION: la regione del cluster.
  2. Crea il pool di nodi. Dopo aver creato il cluster, puoi aggiungere un pool di nodi con G2 o G4. Per i workload multi-nodo che utilizzano G2 (L4) o G4 (RTX PRO 6000), consigliamo di utilizzare una policy di posizionamento compatto per ridurre al minimo la latenza di rete tra i nodi.

    Per creare un pool di nodi, utilizza il seguente comando:

    G2 (NVIDIA L4)

    gcloud container node-pools create NODE_POOL_NAME \
        --cluster=CLUSTER_NAME \
        --region=REGION \
        --node-locations=ZONE \
        --machine-type=MACHINE_TYPE \
        --accelerator=type=nvidia-l4,count=AMOUNT,gpu-driver-version=LATEST \
        --placement-type=COMPACT \
        --scopes="https://www.googleapis.com/auth/cloud-platform" \
        --local-ssd-count=LOCAL_SSD_COUNT
    

    G4 (NVIDIA RTX PRO 6000)

    gcloud container node-pools create NODE_POOL_NAME \
        --cluster=CLUSTER_NAME \
        --region=REGION \
        --node-locations=ZONE \
        --machine-type=MACHINE_TYPE \
        --accelerator=type=nvidia-rtx-pro-6000,count=AMOUNT,gpu-driver-version=LATEST \
        --disk-type=hyperdisk-balanced \
        --placement-type=COMPACT \
        --scopes="https://www.googleapis.com/auth/cloud-platform" \
        --local-ssd-count=LOCAL_SSD_COUNT
    

    Workstation virtuale (vWS) G4

    gcloud container node-pools create NODE_POOL_NAME \
        --cluster=CLUSTER_NAME \
        --region=REGION \
        --node-locations=ZONE \
        --machine-type=MACHINE_TYPE \
        --accelerator=type=nvidia-rtx-pro-6000-vws,count=AMOUNT,gpu-driver-version=LATEST \
        --disk-type=hyperdisk-balanced \
        --placement-type=COMPACT \
        --scopes="https://www.googleapis.com/auth/cloud-platform" \
        --local-ssd-count=LOCAL_SSD_COUNT
    

    Sostituisci quanto segue:

    • NODE_POOL_NAME: il nome del pool di nodi.
    • CLUSTER_NAME: il nome del tuo cluster.
    • REGION: la regione del cluster.
    • ZONE: una o più zone all'interno della tua regione in cui sono disponibili le GPU richieste, ad esempio us-central1-a. Questo è necessario quando si utilizza il posizionamento compatto.
    • MACHINE_TYPE: il tipo di macchina per i nodi, ad esempio g2-standard-4 per le macchine G2, g4-standard-48 per una macchina G4 con una sola GPU o g4-standard-6, g4-standard-12 o g4-standard-24 per i tipi di macchine G4 con meno di una GPU (con il tipo di acceleratore nvidia-rtx-pro-6000).
    • AMOUNT: il numero di GPU da collegare a ogni nodo. Se utilizzi un tipo di macchina G4 con meno di una GPU (g4-standard-6, g4-standard-12 o g4-standard-24) o g4-standard-48 (una GPU), devi impostare AMOUNT su 1.
    • LOCAL_SSD_COUNT: il numero di volumi SSD locali da provisionare su ogni nodo. Ometti il flag --local-ssd-count se utilizzi il tipo di macchina g4-standard-6, perché g4-standard-6 non supporta gli SSD locali.

Connessione al tuo cluster

Connettiti al cluster per poter eseguire i comandi kubectl nelle sezioni successive:

gcloud container clusters get-credentials CLUSTER_NAME \
    --region=REGION

Sostituisci quanto segue:

  • CLUSTER_NAME: il nome del tuo cluster.
  • REGION: la regione del cluster.

Per saperne di più, consulta Installare kubectl e configurare l'accesso al cluster.

Configura i manifest dei pod (solo Autopilot)

Se hai creato un cluster Autopilot, devi selezionare le GPU appropriate nei manifest dei pod in modo che GKE esegua il provisioning dell'hardware.

  1. Specifica il tipo di GPU scelto e la prenotazione specifica utilizzando i selettori di nodi:

    spec:
      nodeSelector:
        cloud.google.com/gke-accelerator: ACCELERATOR
        cloud.google.com/gke-gpu-driver-version: latest
        # Optional: Include if using reserved capacity
        cloud.google.com/reservation-name: RESERVATION_NAME
        cloud.google.com/reservation-affinity: "specific"
    

    Sostituisci quanto segue:

    • ACCELERATOR: l'acceleratore che hai prenotato. Devi utilizzare nvidia-l4 (per G2), nvidia-rtx-pro-6000 (per G4) o nvidia-rtx-pro-6000-vws (per G4 con workstation virtuale).
    • RESERVATION_NAME: il nome della prenotazione di capacità di Compute Engine. Per utilizzare le prenotazioni condivise o blocchi e sottoblocchi specifici di prenotazioni, consulta le sezioni pertinenti in Utilizzo di risorse di percorso zonale riservate.
  2. Aggiungi le seguenti risorse al container che richiede le GPU:

    containers:
      - name: my-container
        resources:
          limits:
            nvidia.com/gpu: AMOUNT
    

    Sostituisci AMOUNT con il numero di GPU che il container deve utilizzare. Per richiedere un tipo di macchina G4 con meno di una GPU (g4-standard-6, g4-standard-12 o g4-standard-24) in un cluster Autopilot, devi utilizzare un ComputeClass personalizzato che specifichi il tipo di macchina e devi impostare nvidia.com/gpu su 1. Per istruzioni, vedi Richiedere tipi di macchina G4 con meno di una GPU.

Passaggi successivi