Container predefiniti per l'addestramento personalizzato

Vertex AI fornisce immagini di container Docker che esegui come container predefiniti per l'addestramento personalizzato. Questi container, organizzati per framework di machine learning (ML) e versione del framework, includono dipendenze comuni che potresti voler utilizzare nel codice di addestramento. In molti casi, l'utilizzo di un container predefinito è più semplice rispetto alla creazione di un container personalizzato per l'addestramento.

Questo documento elenca i container predefiniti per l'addestramento e descrive come utilizzarli con un'applicazione di addestramento Python.

Immagini container disponibili

Ognuna delle seguenti immagini container è disponibile in diversi repository Artifact Registry, che memorizzano i dati in varie posizioni. Puoi utilizzare uno qualsiasi degli URI per un'immagine quando esegui l'addestramento personalizzato; ognuno fornisce la stessa immagine container. Se utilizzi Google Cloud Console per eseguire l'addestramento personalizzato, Google Cloud Console selezionerà l'URI che meglio corrisponde alla località in cui utilizzi Vertex AI per ridurre la latenza.

TensorFlow

Versione framework ML Acceleratori supportati (e versione CUDA, se applicabile) URI (scegline uno) Dipendenze incluse
2,8 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
Pacchetti PyPI Pacchetti Ubuntu
TensorFlow ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
libio-all-perl
libyaml-0-2
Altri Deep Learning Containers
2,8 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
Pacchetti PyPI Pacchetti Ubuntu
TensorFlow ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
libio-all-perl
libyaml-0-2
Altri Deep Learning Containers
2,8 TPU
  • us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-8:latest
Pacchetti PyPI Pacchetti Ubuntu
TensorFlow ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
libio-all-perl
libyaml-0-2
Altri Deep Learning Containers
2.7 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
Pacchetti PyPI Pacchetti Ubuntu
tensorflow 2.7.x (ultima patch)
sympy 1.9
xgboost 1.5.0
python-json-logger 2.0.2
WebOb 1.8.7
Incolla 3.5.0
webapp2 3.0.0b1
fittizi 4.0.3
20
f


2.7 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
Pacchetti PyPI Pacchetti Ubuntu
tensorflow 2.7.x (ultima patch)
sympy 1.9
xgboost 1.5.0
python-json-logger 2.0.2
WebOb 1.8.7
Incolla 3.5.0
webapp2 3.0.0b1
fittizi 4.0.3
20
f


2,6 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
Pacchetti PyPI Pacchetti Ubuntu
TensorFlow f


09
2,6 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
Pacchetti PyPI Pacchetti Ubuntu
TensorFlow f


2,5 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
Pacchetti PyPI Pacchetti Ubuntu
tensorflow 2.5.x (ultima patch)
sympy 1.8
xgboost 1.4.0
PyYAML 3.13
python-json-logger 2.0.1
WebOb 1.8.7
Incolla 3.5.0
webapp2 3.0.0b
-
f


09
2,5 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
Pacchetti PyPI Pacchetti Ubuntu
tensorflow 2.5.x (ultima patch)
sympy 1.8
xgboost 1.4.0
PyYAML 3.13
python-json-logger 2.0.1
WebOb 1.8.7
Incolla 3.5.0
webapp2 3.0.0b
-
f


2,4 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
Pacchetti PyPI Pacchetti Ubuntu
(2) accanto a
, attiva l'operazione anche eseguendo un'operazione, modifica il modo in cui puoi cambiare (fornisci un'operazione) eseguendo l'operazione, modifica l'operazione eseguendo un'operazione forte (fornisci un'operazione) eseguendo un'operazione in modo forte (fornisci un'operazione) diventerà un servizio
f


09
2,4 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
Pacchetti PyPI Pacchetti Ubuntu
(2) accanto a
, attiva l'operazione anche eseguendo un'operazione, modifica il modo in cui puoi cambiare (fornisci un'operazione) eseguendo l'operazione, modifica l'operazione eseguendo un'operazione per applicare un'operazione nei casi in cui l'operazione venga aggiornata eseguendo l'operazione l'operazione diventerà forte (l'operazione sarà necessaria)
f


09
2,3 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
Pacchetti PyPI Pacchetti Ubuntu
(2) trova 2. [0] dà vita con l'operazione troverai infatti l'operazione perfetta con l'operazione che riuscirai a cambiare versione (fornisci) eseguendo un'operazione in attesa) effettua un'operazione (con un forte numero di ricari di un'operazione) f


09
2,3 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
Pacchetti PyPI Pacchetti Ubuntu
(2) trova 2. [0] dà vita con l'operazione troverai infatti l'operazione perfetta con l'operazione che riuscirai a cambiare versione (fornisci) eseguendo un'operazione in attesa) effettua un'operazione (con un forte numero di ricari di un'operazione) f


2.2 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
Pacchetti PyPI Pacchetti Ubuntu
L'opzione 0 da' eseguire con questo ordine a eseguire l'acquisto o quanto sia Quando Quando esegui tu' Quando eseguire a forza di quali si personali anche Quando si verifichi'... Quando vengono applicate licenze di Quando da utilizzare a... Quando scegli di attivare l'applicazione/... ageni anche con questo ordine/ si. anche esegui/ non esegui a elementi a attivare a ordine a comunicare a comunicare con [aiI/ fori/aperi o aVerifica con... ai’ a esegui vengono esegui/ determina/ tue a ordine o a ordine a comunicare con caso a ordine con caso a più a... tua attività ad eseguire ordine, ad ordine a a ordine vengono determinate unità in Quando si eseguirà a mostrati a Collega con Kit. f


2.2 GPU (CUDA 10.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
Pacchetti PyPI Pacchetti Ubuntu
L'opzione 0 da' eseguire con questo ordine a eseguire l'acquisto o quanto sia Quando Quando esegui tu' Quando eseguire a forza di quali si personali anche Quando si verifichi'... Quando vengono applicate licenze di Quando da utilizzare a... Quando scegli di attivare l'applicazione/... ageni anche con questo ordine/ si. anche esegui/ non esegui a elementi a attivare a ordine a comunicare a comunicare con [aiI/ fori/aperi o aVerifica con... ai’ a esegui vengono esegui/ determina/ tue a ordine o a ordine a comunicare con caso a ordine con caso a più a... tua attività ad eseguire ordine, ad ordine a a ordine vengono determinate unità in Quando si eseguirà a mostrati a Collega con Kit. f


2.1 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
Pacchetti PyPI Pacchetti Ubuntu
L'abbonamento a una troverai anche in caso di email con un'email o a un genio che dà luogo a una comunicazione con un buon ricapo di una determinata ricompensa. f


2.1 GPU (CUDA 10.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
Pacchetti PyPI Pacchetti Ubuntu
L'abbonamento a una troverai anche in caso di email con un'email o a un genio che dà luogo a una comunicazione con un buon ricapo di una determinata ricompensa. f


09
1,15 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
Pacchetti PyPI Pacchetti Ubuntu
L'utente verrà eliminato con la ricorrenza con un ricorrenza: il tuo genio sarà eliminato ed riuscirai a farlo. f


09
1,15 GPU (CUDA 10.x)
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest
Pacchetti PyPI Pacchetti Ubuntu
L'utente verrà eliminato con la ricorrenza con un ricorrenza: il tuo genio sarà eliminato ed riuscirai a farlo. f


scikit-learn

Versione framework ML Usare con le GPU? URI (scegline uno) Dipendenze incluse
0,23 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest
  • europe-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest
  • asia-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest
Pacchetti PyPI Pacchetti Ubuntu
175 f


09

PyTorch

Versione framework ML Usare con le GPU? URI (scegline uno) Dipendenze incluse
1.11 No
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
Pacchetti PyPI Pacchetti Ubuntu
PyTorch 1.11.x (ultima patch)
absl-py 1.0.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Altro Dipendenze Deep Learning Containers
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
libio-all-perl
libyaml-0-2
1.11 Sì (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
Pacchetti PyPI Pacchetti Ubuntu
PyTorch 1.11.x (ultima patch)
absl-py 1.0.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Altro Dipendenze Deep Learning Containers
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
libio-all-perl
libyaml-0-2
1.10 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
Pacchetti PyPI Pacchetti Ubuntu
PyTorch 1.10.x (ultima patch)
absl-py 1.0.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Altro Dipendenze Deep Learning Containers
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
libio-all-perl
libyaml-0-2
1.10 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
Pacchetti PyPI Pacchetti Ubuntu
PyTorch 1.10.x (ultima patch)
absl-py 1.0.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Altro Dipendenze Deep Learning Containers
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
libio-all-perl
libyaml-0-2
1.9 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
Pacchetti PyPI Pacchetti Ubuntu
PyTorch 1.9.x (ultima patch)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Altro Dipendenze Deep Learning Containers
f


09
1.9 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
Pacchetti PyPI Pacchetti Ubuntu
PyTorch 1.9.x (ultima patch)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Altro Dipendenze Deep Learning Containers
f


09
1,7 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
Deep dipendenze dei container per l'apprendimento
1,7 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
Deep dipendenze dei container per l'apprendimento
1.6 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
Deep dipendenze dei container per l'apprendimento
1.6 GPU (CUDA 11.x)
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
Deep dipendenze dei container per l'apprendimento
1.4 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
Deep dipendenze dei container per l'apprendimento
1.4 GPU (CUDA 10.x)
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest
Deep dipendenze dei container per l'apprendimento

XGBoost

Versione framework ML Usare con le GPU? URI (scegline uno) Dipendenze incluse
1.1 Nessuna
  • us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest
Pacchetti PyPI Pacchetti Ubuntu
175 f


Addestra con un container predefinito

Per utilizzare un contenitore predefinito, leggi la guida alla configurazione delle impostazioni dei contenitori.

Se utilizzi un'immagine container che supporta le GPU, assicurati di specificare i campi machineSpec.acceleratorType e machineSpec.acceleratorCount quando crei la risorsa CustomJob, HyperparameterTuningJob o TrainingPipeline.

Passaggi successivi