カスタム トレーニング用の事前にビルドされたコンテナ

Vertex AI によって、カスタム トレーニング用のビルド済みコンテナとして実行する Docker コンテナ イメージが提供されます。これらのコンテナは、機械学習(ML)フレームワークとフレームワークのバージョンで編成され、トレーニング コードで使用する共通の依存関係が含まれています。多くの場合、トレーニング用の独自のカスタム コンテナを作成するよりも、ビルド済みコンテナを使用するほうが簡単です。

このドキュメントでは、トレーニング用のビルド済みコンテナを一覧表示し、Python トレーニング アプリケーションで使用する方法について説明します。

使用可能なコンテナ イメージ

次の各コンテナ イメージは、さまざまなロケーションにデータを格納する複数の Artifact Registry リポジトリで利用できます。カスタム トレーニングを行うときは、任意の画像の URI が使用可能であり、各 URI は同じコンテナ イメージを指定しています。Google Cloud Console を使用してカスタム トレーニングを行う場合、Cloud Console は、レイテンシを低減するために、Vertex AI を使用しているロケーションに最適な URI を選択します。

TensorFlow

ML フレームワークのバージョン GPU 対応の有無(CUDA バージョン) URI(任意で選択) 含まれる依存関係
2.6 ×
  • 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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.6.x(最新パッチ)
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
その他の Deep Learning Containers の依存関係
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 ○(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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.6.x(最新パッチ)
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
その他の Deep Learning Containers の依存関係
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 ×
  • 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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.5.x(最新パッチ)
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
その他の Deep Learning Containers の依存関係
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 ○(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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.5.x(最新パッチ)
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
その他の Deep Learning Containers の依存関係
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 ×
  • 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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.4.x(最新のパッチ)
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 ○(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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.4.x(最新のパッチ)
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 ×
  • 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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.3.x(最新のパッチ)
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 ○(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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.3.x(最新のパッチ)
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 ×
  • 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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.2.x(最新のパッチ)
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 ○(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
PyPI パッケージ Ubuntu パッケージ
tensorflow 2.2.x(最新のパッチ)
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 ×
  • 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
PyPI パッケージ 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 ○(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
PyPI パッケージ 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 ×
  • 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
PyPI パッケージ 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 ○(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
PyPI パッケージ 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

ML フレームワークのバージョン GPU 対応の有無 URI(任意で選択) 含まれる依存関係
0.23 ×
  • 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
PyPI パッケージ 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

ML フレームワークのバージョン GPU 対応の有無 URI(任意で選択) 含まれる依存関係
1.9 ×
  • 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
PyPI パッケージ Ubuntu パッケージ
PyTorch 1.9.x(最新パッチ)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
その他の Deep Learning Containers の依存関係
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 ○(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
PyPI パッケージ Ubuntu パッケージ
PyTorch 1.9.x(最新パッチ)
absl-py 0.13.0
cloud-tpu-client 0.10
python-json-logger 2.0.2
cloudml-hypertune
その他の Deep Learning Containers の依存関係
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 ×
  • 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 Learning Containers の依存関係
1.7 ○(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 Learning Containers の依存関係
1.6 ×
  • 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 Learning Containers の依存関係
1.6 ○(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 Learning Containers の依存関係
1.4 ×
  • 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 Learning Containers の依存関係
1.4 ○(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 Learning Containers の依存関係

XGBoost

ML フレームワークのバージョン GPU 対応の有無 URI(任意で選択) 含まれる依存関係
1.1 ×
  • 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
PyPI パッケージ 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

ビルド済みのコンテナを使用してトレーニングする

ビルド済みコンテナを使用するには、コンテナ設定の構成に関するガイドをご覧ください。

GPU をサポートするコンテナ イメージを使用している場合は、CustomJobHyperparameterTuningJobTrainingPipeline の各リソース作成時に machineSpec.acceleratorTypemachineSpec.acceleratorCount のフィールドを指定してください。

次のステップ