Esempio: eseguire operazioni di avvio anticipato con un boothook cloud-init

Fornisci questo script come valore "userData" nel modello di avvio. Per ulteriori informazioni, consulta Lavorare con i dati utente di Amazon EC2 per AWS 2 PEZZI.

Alcune configurazioni devono essere eseguite all'inizio della sequenza di avvio, prima dell'avvio del bootstrap del AWS PCS e prima che qualsiasi servizio utilizzi l'hardware. I casi tipici sono i comandi di configurazione della GPU che non possono avere effetto mentre i processi mantengono le GPU, come l'attivazione della modalità NVIDIA Multi-Instance GPU (MIG), e le operazioni che richiedono un riavvio per avere effetto, come la modifica della modalità GPU ECC. Un riavvio in questa fase è sicuro, perché il bootstrap del PC non è ancora iniziato. AWS Non riavviate più avanti nel processo di avvio, ad esempio da uno script di azione del ciclo di vita del nodo: un riavvio interrompe la sequenza di bootstrap del PCS. AWS Per ulteriori informazioni, consulta Le migliori pratiche per le azioni del ciclo di vita dei nodi in AWS PEZZI.

Per creare l'hook di avvio, aggiungete un blocco con il tipo di contenuto ai dati utente. text/cloud-boothook Il blocco contiene lo script stesso. Un boothook è il primo hook disponibile nei dati utente: lo cloud-init esegue non appena elabora il blocco, prima di cloud-config direttive come bootcmd e prima dei blocchi di dati utente del AWS PCS che registrano il nodo. Un boothook viene eseguito ad ogni avvio, incluso il riavvio che potrebbe innescare, quindi lo script deve essere idempotente: controllate prima lo stato corrente e saltate il lavoro che è già stato fatto. Per ulteriori informazioni, consulta la documentazione su cloud-init.

L'esempio seguente abilita la modalità MIG su tutte le GPU e crea le istanze MIG a ogni avvio. Poiché nessun client GPU è ancora stato avviato, la modifica della modalità ha effetto immediato e non è necessario riavviare.

MIME-Version: 1.0
Content-Type: multipart/mixed; boundary="==MYBOUNDARY=="

--==MYBOUNDARY==
Content-Type: text/cloud-boothook; charset="us-ascii"

#!/bin/bash
if ! nvidia-smi -i 0 -q | grep -A1 'MIG Mode' | grep -q 'Current.*Enabled'; then
    nvidia-smi -mig 1
fi
if [ -z "$(nvidia-smi -L | grep MIG)" ]; then
    for i in 0 1 2 3 4 5 6 7; do
        nvidia-smi mig -i $i -cgi 9,14,19,19 -C
    done
fi

--==MYBOUNDARY==--

Se l'operazione indica che è necessario un riavvio, riavvia dallo stesso boothook dopo il cambio di stato. Al successivo avvio, i controlli di idempotenza rilevano il lavoro già svolto, quindi il nodo prosegue normalmente nel bootstrap del PC. AWS