Riferimento CTPU

Panoramica

Lo strumento ctpu open source viene utilizzato per creare un flock di risorse di calcolo, costituite da una VM Compute Engine e uno o più dispositivi Cloud TPU. Lo strumento è preinstallato in Cloud shell.

Puoi trovare documentazione e codice per ctpu su GitHub.

Lo strumento ctpu utilizza la seguente sintassi:

ctpu <subcommand> <flags> <subcommand> <subcommand args> 

Di seguito sono riportati i sottocomandi per ctpu:

auth

Descrizione
Imposta o mostra le autorizzazioni per le Cloud TPU.
Utilizzo
ctpu auth <flags> <subcommand> <subcommand args>
Esempio
ctpu auth list --project="my-project" --zone=us-central1-a
ctpu auth list --project my-project --zone us-central1-a
Subcomandi

Il comando ctpu auth supporta i seguenti sottocomandi:

  • add-bigtable: assicurati che Cloud TPU sia autorizzato per Cloud Bigtable
  • add-gcs: assicurati che Cloud TPU sia autorizzato per Cloud Storage
  • list: mostra le autorizzazioni dell'account di servizio Cloud TPU
  • commands: elenca tutti i nomi dei comandi
  • flags: descrive tutti i flag di primo livello noti
  • help: descrive i sottocomandi e la relativa sintassi
Flag facoltativi

Di seguito sono riportati i comandi facoltativi per ctpu auth. name | project | zone

elimina (rm)

Descrizione
Elimina la VM di Compute Engine e la Cloud TPU.
Utilizzo
ctpu rm <flags>
Esempio
ctpu rm --zone=us-central1-b

help

Descrizione
Elenca tutti i comandi secondari ctpu e i flag di primo livello.
Utilizzo
ctpu help
ctpu help <subcommand>
Esempio
ctpu help   // list all ctpu subcommands and top level flags

ctpu help auth   // list all flags that can be used with `ctpu auth`
ctpu help up   // list all flags that can be used with `ctpu up`

list (ls)

Descrizione
Elenca tutte le VM Compute Engine e le Cloud TPU nella zona specificata.
Utilizzo
ctpu ls <flags>
Esempio
ctpu ls --zone=us-central1-b

pause (zz)

Descrizione

Arresta la VM di Compute Engine ed elimina la Cloud TPU. Interrompi l'addebito per l'utilizzo di Cloud TPU finché non esegui ctpu up.

Per assicurarti che la Cloud TPU sia arrestata, devi specificare il nome e la zona della Cloud TPU sulla riga di comando.

Utilizzo
ctpu pause <name, zone>
Esempio
ctpu pause --name=my-tpu --zone=us-central1-a  // pause the named TPU
in the specified zone
Descrizione
Stampa sullo schermo la configurazione corrente del nome, del progetto e della zona di Cloud TPU.
Esempio
ctpu print-config

quota

Descrizione
Visualizza un URL in cui puoi visualizzare le quote.
Utilizzo
ctpu quota
Esempio
ctpu quota
Output: Quotas cannot currently be displayed within ctpu.
To view your quota, open <url>
Request additional quota from <url>

restart

Descrizione

Riavvia una Cloud TPU che è ancora in stato RUNNING (indicata in ctpu status), ma la cui esecuzione è stata interrotta a causa di un problema hardware. Utilizza gcloud compute tpu start o il pulsante Avvvia nella pagina Compute Engine > TPU della console Cloud se la TPU è nello stato STOPPED.

restart non riavvia una Cloud TPU prerilasciata. Devi eseguire ctpu delete e ctpu up se la tua Cloud TPU è stata prerilasciata.

Utilizzo
ctpu restart <flags>
Esempio
ctpu restart --zone=us-central1-a

status (st)

Descrizione

Esegui query sulle API Google Cloud (solo zona predefinita) per determinare lo stato attuale della Cloud TPU e della VM Compute Engine.

Utilizzo

ctpu st

Esempio
ctpu st --zone=us-central1-a
Status message:
  Your cluster is running!
    Compute Engine VM:  RUNNING
    Cloud TPU:     RUNNING 

tpu-locations

Descrizione
Elenca tutte le zone in cui sono disponibili i tipi di TPU.
Utilizzo
ctpu tpu-locations
Output
Cloud TPU Locations:
    asia-east1-c
    europe-west4-a
    us-central1-a
    us-central1-b
    us-central1-c

tpu-sizes

Descrizione
Elenca tutte le dimensioni TPU disponibili nella zona specificata. Alcune dimensioni sono disponibili solo in determinate zone. (default = default zone)
Utilizzo
tpu-sizes <zone>
Esempio
ctpu tpu-sizes --zone=us-central1-a

in alto

Descrizione

Visualizza un set di risorse ctpu. La prima volta che esegui ctpu up in un progetto, l'operazione richiede più tempo rispetto alle esecuzioni future perché viene eseguita attività come la propagazione delle chiavi SSH e l'attivazione delle API.

  • Abilita i servizi Compute Engine e Cloud TPU.
  • Crea una VM Compute Engine con la versione stabile più recente di TensorFlow preinstallata.
  • Assegna una zona predefinita, ad esempio us-central1-b, in base alla tua posizione.
  • Passa il nome della Cloud TPU alla VM di Compute Engine come variabile di ambiente (TPU_NAME).
  • Assicurati che Cloud TPU abbia accesso alle risorse di cui ha bisogno dal tuo progetto Google Cloud concedendo ruoli IAM specifici al tuo account di servizio Cloud TPU.
  • Esegue una serie di altri controlli.
  • Accedi alla nuova VM Compute Engine. Il prompt della shell cambia da username@project a username@tpuname.

Puoi eseguire ctpu up tutte le volte che vuoi. Ad esempio, se perdi la connessione SSH alla VM Compute Engine, esegui ctpu up per ripristinarla. Devi specificare una zona se Compute Engine non si trova nella zona predefinita. Ad esempio:

$ ctpu up --zone=us-central1-a
Utilizzo
ctpu up <flags>
Esempio
ctpu up --tpu-size=v2-8 --disk-size-gb=320 --preemptible
Flag

--disk-size-gb
Configura le dimensioni del volume principale della VM Compute Engine. Il valore deve essere un numero intero. (valore predefinito = 250)

--dry-run
Non apportare modifiche; stampa solo ciò che sarebbe successo.

--forward-agent
Abilita l'inoltro dell'agente SSH quando esegui l'accesso SSH alla VM Compute Engine. Il forwarding dell'agente SSH consente di accedere ai repository condivisi (ad esempio GitHub) senza dover inserire le chiavi private sulla VM Compute Engine. (valore predefinito = true)

--forward-ports
Inoltra automaticamente le porte utili dalla VM Compute Engine alla tua macchina locale. Le porte inoltrate sono: 6006 (TensorBoard), 8888 (notebook Jupyter), 8470 (porta TPU), 8466 (porta del profiler TPU). (valore predefinito = true)

--gce-image
Sostituisci l'immagine Compute Engine scelta automaticamente. Utilizza questo flag se utilizzi le tue immagini personalizzate anziché quelle fornite con la versione di TensorFlow installata.

--gcp-network
Specifica la rete in cui devono essere creati Cloud TPU e la VM associata. Per informazioni sulle reti, consulta la panoramica della rete Virtual Private Cloud (VPC). (default = rete predefinita)

--log-http
Stampa i contenuti completi delle coppie di richieste e risposte HTTP. Per attivare la stampa, imposta questo flag su true. Utilizza questo flag quando hai bisogno dell'output del log per inviare una segnalazione di bug relativa a ctpu. Per ulteriori dettagli, consulta il ctpu README.

--machine-type
Configura le dimensioni della VM di Compute Engine. Un elenco completo dei tipi di macchine è disponibile nella pagina Tipi di macchine di Cloud. (valore predefinito = n1-standard-2)

--name
Sostituisci il nome da utilizzare per le VM e Cloud TPU. (valore predefinito = il tuo nome utente)

--noconf
Salta la conferma.

--preemptible
Crea un nodo Cloud TPU prerilasciabile. Una Cloud TPU prerilasciabile costa meno all'ora rispetto a una non prerilasciabile. Il servizio Cloud TPU può uscire da un dispositivo prerilasciabile in qualsiasi momento. (valore predefinito = non prelevabile)

--preemptible-vm
Crea una VM Compute Engine preassegnabile. Una VM prerilasciabile costa meno all'ora rispetto a una VM non prerilasciabile. Il servizio Compute Engine può uscire dall'istanza VM in qualsiasi momento. (valore predefinito = non prelevabile)

--print-welcome
Stampa sempre il messaggio di benvenuto.

--project
Sostituisci il nome del progetto Google Cloud da utilizzare per l'allocazione di VM e TPU. Specifica un valore dai metadati di Cloud Config o Compute Engine, in genere il nome del progetto. Se non è possibile trovare un valore valido, devi fornire un valore sulla riga di comando.

--tf-version
Imposta la versione di TensorFlow da utilizzare per la creazione della VM Compute Engine e di Cloud TPU. (valore predefinito = ultima release stabile)

--tpu-only
Alloca solo una Cloud TPU. Utilizza questa opzione solo se hai già una VM disponibile.

--tpu-size
Configura le dimensioni e la versione hardware di una Cloud TPU.

--use-dl-images
Utilizza le immagini VM per il deep learning (consulta la documentazione all'indirizzo https://cloud.google.com/deep-learning-vm/) anziché le immagini VM TPU. (valore predefinito = immagini macchina TPU)

--vm-only
Alloca solo una VM; utilizza questa opzione se non vuoi configurare e pagare una TPU.

--zone
Sostituisci la zona Compute Engine da utilizzare per l'allocazione di VM e Cloud TPU. Nella riga di comando, esegui ctpu help up per visualizzare l'elenco.

versione

Descrizione
Stampa la versione di ctpu installata.
Utilizzo
ctpu version
Output
ctpu version
Output: ctpu version: 1.9