Conteneurs préconfigurés pour l'entraînement personnalisé

Vertex AI fournit des images de conteneurs Docker que vous exécutez en tant que conteneurs prédéfinis pour l'entraînement personnalisé. Ces conteneurs, organisés par framework de machine learning (ML) et par version de framework, incluent des dépendances courantes que vous pouvez utiliser dans du code d'entraînement. Dans de nombreux cas, l'utilisation d'un conteneur prédéfini est plus simple que de créer un conteneur personnalisé pour l'entraînement.

Ce document répertorie les conteneurs prédéfinis pour l'entraînement et décrit comment les utiliser avec une application d'entraînement Python.

Images de conteneurs disponibles

Chacune des images de conteneur suivantes est disponible dans plusieurs dépôts Artifact Registry qui stockent des données dans différents emplacements. Vous pouvez utiliser n'importe quel URI d'une image lorsque vous effectuez un entraînement personnalisé, chacun pointe vers la même image de conteneur. Si vous utilisez Google Cloud Console pour effectuer un entraînement personnalisé, Cloud Console sélectionne l'URI qui correspond le mieux à l'emplacement où vous utilisez Vertex AI afin de réduire la latence.

TensorFlow

Version du framework de ML Utilisation avec des GPU ? (Version CUDA) URI (choisissez une valeur) Dépendances incluses
2,6 Non
  • 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
Packages PyPI Packages Ubuntu
tensorflow 2.6.x (latest patch)
sympy 1.8
xgboost 1.4.2
python-json-logger 2.0.2
WebOb 1.8.7
Paste 3.5.0
webapp2 3.0.0b1
mock 4.0.3
google-cloud-resource-manager 1.0.2
cloudml-hypertune
Other Deep Learning Containers dependencies
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2,6 Oui (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
Packages PyPI Packages Ubuntu
tensorflow 2.6.x (latest patch)
sympy 1.8
xgboost 1.4.2
python-json-logger 2.0.2
WebOb 1.8.7
Paste 3.5.0
webapp2 3.0.0b1
mock 4.0.3
google-cloud-resource-manager 1.0.2
cloudml-hypertune
Other Deep Learning Containers dependencies
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.5 Non
  • 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
Packages PyPI Packages Ubuntu
tensorflow 2.5.x (dernier correctif)
sympy 1.8
xgboost 1.4.0
PyYAML 3.13
python-json-logger 2.0.1
WebOb 1.8.7
Paste 3.5.0
webapp2 3.0.0b1
mock 4.0.3
google-cloud-resource-manager 0.30.3
cloudml-hypertune
Autres dépendances de conteneurs de deep learning
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.5 Oui (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
Packages PyPI Packages Ubuntu
tensorflow 2.5.x (dernier correctif)
sympy 1.8
xgboost 1.4.0
PyYAML 3.13
python-json-logger 2.0.1
WebOb 1.8.7
Paste 3.5.0
webapp2 3.0.0b1
mock 4.0.3
google-cloud-resource-manager 0.30.3
cloudml-hypertune
Autres dépendances de conteneurs de deep learning
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.4 Non
  • 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
Packages PyPI Packages Ubuntu
tensorflow 2.4.x (latest patch)
numpy 1.19.4
pandas 1.1.5
scipy 1.5.4
scikit-learn 0.24.0
sympy 1.7.1
statsmodels 0.12.1
xgboost 1.3.1
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 3.13
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.0
python-json-logger 2.0.1
wheel 0.36.2
WebOb 1.8.6
Paste 3.5.0
tornado 5.1.1
grpcio 1.32.0
requests 2.25.1
webapp2 3.0.0b1
mock 4.0.3
google-cloud-bigquery 2.6.1
google-cloud-bigtable 1.6.1
google-cloud-datastore 2.1.0
google-cloud-logging 1.15.0
google-cloud-pubsub 2.2.0
google-cloud-resource-manager 0.30.3
google-cloud-storage 1.35.0
joblib 1.0.0
cloudml-hypertune
psutil 5.8.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.4 Oui (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
Packages PyPI Packages Ubuntu
tensorflow 2.4.x (latest patch)
numpy 1.19.4
pandas 1.1.5
scipy 1.5.4
scikit-learn 0.24.0
sympy 1.7.1
statsmodels 0.12.1
xgboost 1.3.1
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 3.13
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.0
python-json-logger 2.0.1
wheel 0.36.2
WebOb 1.8.6
Paste 3.5.0
tornado 5.1.1
grpcio 1.32.0
requests 2.25.1
webapp2 3.0.0b1
mock 4.0.3
google-cloud-bigquery 2.6.1
google-cloud-bigtable 1.6.1
google-cloud-datastore 2.1.0
google-cloud-logging 1.15.0
google-cloud-pubsub 2.2.0
google-cloud-resource-manager 0.30.3
google-cloud-storage 1.35.0
joblib 1.0.0
cloudml-hypertune
psutil 5.8.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2,3 Non
  • 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
Packages PyPI Packages Ubuntu
tensorflow 2.3.x (latest patch)
numpy 1.18.5
pandas 1.1.3
scipy 1.4.1
scikit-learn 0.23.2
sympy 1.6.2
statsmodels 0.12.0
xgboost 1.2.1
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 3.13
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.0
python-json-logger 2.0.1
wheel 0.35.1
WebOb 1.8.6
Paste 3.5.0
tornado 5.1.1
grpcio 1.33.1
requests 2.24.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 2.2.0
google-cloud-bigtable 1.5.1
google-cloud-datastore 1.15.3
google-cloud-logging 1.15.0
google-cloud-pubsub 2.1.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.32.0
joblib 0.17.0
cloudml-hypertune
psutil 5.7.2
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2,3 Oui (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
Packages PyPI Packages Ubuntu
tensorflow 2.3.x (latest patch)
numpy 1.18.5
pandas 1.1.3
scipy 1.4.1
scikit-learn 0.23.2
sympy 1.6.2
statsmodels 0.12.0
xgboost 1.2.1
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 3.13
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.0
python-json-logger 2.0.1
wheel 0.35.1
WebOb 1.8.6
Paste 3.5.0
tornado 5.1.1
grpcio 1.33.1
requests 2.24.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 2.2.0
google-cloud-bigtable 1.5.1
google-cloud-datastore 1.15.3
google-cloud-logging 1.15.0
google-cloud-pubsub 2.1.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.32.0
joblib 0.17.0
cloudml-hypertune
psutil 5.7.2
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.2 Non
  • 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
Packages PyPI Packages Ubuntu
tensorflow 2.2.x (latest patch)
numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 0.23.1
sympy 1.6
statsmodels 0.11.1
xgboost 1.1.1
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune
psutil 5.7.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.2 Oui (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
Packages PyPI Packages Ubuntu
tensorflow 2.2.x (latest patch)
numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 0.23.1
sympy 1.6
statsmodels 0.11.1
xgboost 1.1.1
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune
psutil 5.7.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.1 Non
  • 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
Packages PyPI Packages Ubuntu
tensorflow 2.1.0
numpy 1.18.0
pandas 0.25.3
scipy 1.4.1
scikit-learn 0.22.1
sympy 1.5
statsmodels 0.10.2
xgboost 0.90
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.13.0
future 0.18.2
PyYAML 5.2
wrapt 1.11.2
crcmod 1.7
google-api-python-client 1.7.11
python-json-logger 0.1.11
wheel 0.33.6
WebOb 1.8.5
Paste 3.2.3
tornado 5.1.1
grpcio 1.26.0
requests 2.22.0
webapp2 3.0.0b1
mock 3.0.5
google-cloud-bigquery 1.23.1
google-cloud-bigtable 1.2.0
google-cloud-datastore 1.10.0
google-cloud-logging 1.14.0
google-cloud-pubsub 1.1.0
google-cloud-resource-manager 0.30.0
google-cloud-storage 1.23.0
joblib 0.14.1
cloudml-hypertune
psutil 5.7.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
2.1 Oui (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
Packages PyPI Packages Ubuntu
tensorflow 2.1.0
numpy 1.18.0
pandas 0.25.3
scipy 1.4.1
scikit-learn 0.22.1
sympy 1.5
statsmodels 0.10.2
xgboost 0.90
oauth2client 4.1.3
httplib2 0.15.0
python-dateutil 2.8.1
six 1.13.0
future 0.18.2
PyYAML 5.2
wrapt 1.11.2
crcmod 1.7
google-api-python-client 1.7.11
python-json-logger 0.1.11
wheel 0.33.6
WebOb 1.8.5
Paste 3.2.3
tornado 5.1.1
grpcio 1.26.0
requests 2.22.0
webapp2 3.0.0b1
mock 3.0.5
google-cloud-bigquery 1.23.1
google-cloud-bigtable 1.2.0
google-cloud-datastore 1.10.0
google-cloud-logging 1.14.0
google-cloud-pubsub 1.1.0
google-cloud-resource-manager 0.30.0
google-cloud-storage 1.23.0
joblib 0.14.1
cloudml-hypertune
psutil 5.7.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
1.15 Non
  • 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
Packages PyPI Packages Ubuntu
tensorflow 1.15.0
numpy 1.16.5
pandas 0.24.2
scipy 1.2.2
scikit-learn 0.20.4
sympy 1.4
statsmodels 0.10.1
xgboost 0.82
oauth2client 4.1.3
httplib2 0.12.0
python-dateutil 2.7.5
argparse 1.4.0
six 1.12.0
future 0.17.1
PyYAML 3.13
wrapt 1.11.1
crcmod 1.7
google-api-python-client 1.7.8
python-json-logger 0.1.10
wheel 0.32.3
WebOb 1.8.5
Paste 3.0.6
tornado 5.1.1
grpcio 1.18.0
requests 2.19.0
webapp2 3.0.0b1
mock 2.0.0
google-cloud-bigquery 1.20.0
google-cloud-bigtable 1.0.0
google-cloud-datastore 1.9.0
google-cloud-logging 1.12.1
google-cloud-pubsub 1.0.0
google-cloud-resource-manager 0.29.2
google-cloud-storage 1.19.1
joblib 0.13.0
cloudml-hypertune
psutil 5.7.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
1.15 Oui (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
Packages PyPI Packages Ubuntu
tensorflow 1.15.0
numpy 1.16.5
pandas 0.24.2
scipy 1.2.2
scikit-learn 0.20.4
sympy 1.4
statsmodels 0.10.1
xgboost 0.82
oauth2client 4.1.3
httplib2 0.12.0
python-dateutil 2.7.5
argparse 1.4.0
six 1.12.0
future 0.17.1
PyYAML 3.13
wrapt 1.11.1
crcmod 1.7
google-api-python-client 1.7.8
python-json-logger 0.1.10
wheel 0.32.3
WebOb 1.8.5
Paste 3.0.6
tornado 5.1.1
grpcio 1.18.0
requests 2.19.0
webapp2 3.0.0b1
mock 2.0.0
google-cloud-bigquery 1.20.0
google-cloud-bigtable 1.0.0
google-cloud-datastore 1.9.0
google-cloud-logging 1.12.1
google-cloud-pubsub 1.0.0
google-cloud-resource-manager 0.29.2
google-cloud-storage 1.19.1
joblib 0.13.0
cloudml-hypertune
psutil 5.7.0
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

scikit-learn

Version du framework de ML Utilisation avec des GPU ? URI (choisissez une valeur) Dépendances incluses
0.23 Non
  • 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
Packages PyPI Packages Ubuntu
numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 0.23.1
sympy 1.6
statsmodels 0.11.1
xgboost 1.1.1
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

PyTorch

Version du framework de ML Utilisation avec des GPU ? URI (choisissez une valeur) Dépendances incluses
1.9 Non
  • 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
Packages PyPI Packages Ubuntu
PyTorch 1.9.x (dernier correctif)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Other Autres dépendances de conteneurs de deep learning
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
1.9 Oui (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
Packages PyPI Packages Ubuntu
PyTorch 1.9.x (dernier correctif)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
Other Autres dépendances de conteneurs de deep learning
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv
1,7 Non
  • 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
Dépendances des conteneurs de deep learning
1,7 Oui (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
Dépendances des conteneurs de deep learning
1,6 Non
  • 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
Dépendances des conteneurs de deep learning
1,6 Oui (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
Dépendances des conteneurs de deep learning
1.4 Non
  • 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
Dépendances des conteneurs de deep learning
1.4 Oui (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
Dépendances des conteneurs de deep learning

XGBoost

Version du framework de ML Utilisation avec des GPU ? URI (choisissez une valeur) Dépendances incluses
1.1 Non
  • 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
Packages PyPI Packages Ubuntu
numpy 1.18.5
pandas 1.0.4
scipy 1.4.1
scikit-learn 0.23.1
sympy 1.6
statsmodels 0.11.1
xgboost 1.1.1
oauth2client 4.1.3
httplib2 0.18.1
python-dateutil 2.8.1
six 1.15.0
future 0.18.2
PyYAML 5.3.1
wrapt 1.12.1
crcmod 1.7
google-api-python-client 1.9.3
python-json-logger 0.1.11
wheel 0.34.2
WebOb 1.8.6
Paste 3.4.1
tornado 5.1.1
grpcio 1.29.0
requests 2.23.0
webapp2 3.0.0b1
mock 4.0.2
google-cloud-bigquery 1.25.0
google-cloud-bigtable 1.2.1
google-cloud-datastore 1.12.0
google-cloud-logging 1.15.0
google-cloud-pubsub 1.6.0
google-cloud-resource-manager 0.30.2
google-cloud-storage 1.29.0
joblib 0.15.1
cloudml-hypertune
curl
libcurl3-dev
wget
zip
unzip
git
vim
build-essential
ca-certificates
pkg-config
rsync
ca-certificates-java
libatlas-base-dev
liblapack-dev
gfortran
libpng-dev
libjpeg-dev
python3.7
python3-dev
python3.7-dev
python3-pip
python3-setuptools
python2.7
python-dev
python-setuptools
gdb
openjdk-8-jdk
openjdk-8-jre-headless
g++
zlib1g-dev
libio-all-perl
module-init-tools
libsnappy-dev
libyaml-0-2
python-opencv

Entraînement avec des conteneurs prédéfinis

Pour utiliser un conteneur prédéfini, consultez le guide de configuration des paramètres de conteneur.

Si vous utilisez une image de conteneur compatible avec les GPU, veillez à spécifier les champs machineSpec.acceleratorType et machineSpec.acceleratorCount lorsque vous créez votre ressource CustomJob, HyperparameterTuningJob ou TrainingPipeline.

Étape suivante