このページでは、使用するコンテナ イメージの選択に役立つ情報を提供します。
コンテナ イメージのタイプを選択する
各コンテナ イメージは Python 3 環境を提供します。代表的なデータ サイエンス フレームワーク(PyTorch や TensorFlow など)、Conda、GPU イメージ用 NVIDIA スタック(CUDA、cuDNN、NCCL2)など、多くのサポート パッケージとツールが含まれています。必要なコンテナ イメージについては、下の表をご覧ください。
次の Deep Learning Containers イメージタイプの一覧は、フレームワーク タイプ別に整理されています。
フレームワーク | プロセッサ | コンテナ イメージ名 |
---|---|---|
基本 | GPU |
gcr.io/deeplearning-platform-release/base-cu110 gcr.io/deeplearning-platform-release/base-cu113
|
CPU | gcr.io/deeplearning-platform-release/base-cpu |
|
TensorFlow Enterprise 2.x | GPU | gcr.io/deeplearning-platform-release/tf-gpu gcr.io/deeplearning-platform-release/tf-gpu.2-1 gcr.io/deeplearning-platform-release/tf-gpu.2-3 gcr.io/deeplearning-platform-release/tf-gpu.2-6 gcr.io/deeplearning-platform-release/tf-gpu.2-8 gcr.io/deeplearning-platform-release/tf-gpu.2-9
|
CPU | gcr.io/deeplearning-platform-release/tf-cpu gcr.io/deeplearning-platform-release/tf-cpu.2-1 gcr.io/deeplearning-platform-release/tf-cpu.2-3 gcr.io/deeplearning-platform-release/tf-cpu.2-6 gcr.io/deeplearning-platform-release/tf-cpu.2-8 gcr.io/deeplearning-platform-release/tf-cpu.2-9
|
|
TensorFlow Enterprise 1.x | GPU | gcr.io/deeplearning-platform-release/tf-gpu.1-15
|
CPU | gcr.io/deeplearning-platform-release/tf-cpu.1-15
|
|
PyTorch | GPU | gcr.io/deeplearning-platform-release/pytorch-gpu gcr.io/deeplearning-platform-release/pytorch-gpu.1-11
|
R | CPU(試験運用版) | gcr.io/deeplearning-platform-release/r-cpu gcr.io/deeplearning-platform-release/r-cpu.4-1
|
Scikit-learn | CPU(試験運用版) | gcr.io/deeplearning-platform-release/sklearn-cpu gcr.io/deeplearning-platform-release/sklearn-cpu.0-23
|
XGBoost | CPU(試験運用版) | gcr.io/deeplearning-platform-release/xgboost-cpu gcr.io/deeplearning-platform-release/xgboost-cpu.1-1
|
非推奨のイメージ | 混合 | gcr.io/deeplearning-platform-release/base-cu100 gcr.io/deeplearning-platform-release/base-cu101 gcr.io/deeplearning-platform-release/tf2-gpu.2-0 gcr.io/deeplearning-platform-release/tf2-cpu.2-0 gcr.io/deeplearning-platform-release/tf2-gpu.2-2 gcr.io/deeplearning-platform-release/tf2-cpu.2-2 gcr.io/deeplearning-platform-release/tf2-gpu.2-4 gcr.io/deeplearning-platform-release/tf2-cpu.2-4 gcr.io/deeplearning-platform-release/tf-gpu.2-5 gcr.io/deeplearning-platform-release/tf-gpu.2-7 gcr.io/deeplearning-platform-release/tf-cpu.2-5 gcr.io/deeplearning-platform-release/tf-cpu.2-7 gcr.io/deeplearning-platform-release/tf-gpu.1-13 gcr.io/deeplearning-platform-release/tf-gpu.1-14 gcr.io/deeplearning-platform-release/tf-cpu.1-13 gcr.io/deeplearning-platform-release/tf-cpu.1-14 gcr.io/deeplearning-platform-release/pytorch-gpu.1-0 gcr.io/deeplearning-platform-release/pytorch-gpu.1-1 gcr.io/deeplearning-platform-release/pytorch-gpu.1-2 gcr.io/deeplearning-platform-release/pytorch-gpu.1-3 gcr.io/deeplearning-platform-release/pytorch-gpu.1-4 gcr.io/deeplearning-platform-release/pytorch-gpu.1-6 gcr.io/deeplearning-platform-release/pytorch-gpu.1-7 gcr.io/deeplearning-platform-release/pytorch-gpu.1-8 gcr.io/deeplearning-platform-release/pytorch-gpu.1-9 gcr.io/deeplearning-platform-release/pytorch-gpu.1-10 gcr.io/deeplearning-platform-release/pytorch-cpu.1-0 gcr.io/deeplearning-platform-release/pytorch-cpu.1-1 gcr.io/deeplearning-platform-release/pytorch-cpu.1-2 gcr.io/deeplearning-platform-release/pytorch-cpu.1-3 gcr.io/deeplearning-platform-release/pytorch-cpu.1-4 gcr.io/deeplearning-platform-release/pytorch-xla.1-6 gcr.io/deeplearning-platform-release/pytorch-xla.1-7 gcr.io/deeplearning-platform-release/pytorch-xla.1-8 gcr.io/deeplearning-platform-release/pytorch-xla.1-9 gcr.io/deeplearning-platform-release/pytorch-xla.1-10 gcr.io/deeplearning-platform-release/pytorch-xla.1-11
|
含まれる依存関係
各リリースに含まれる Python 依存関係のリストは、Cloud Storage の次の場所で提供されています。
gs://deeplearning-platform-release/installed-dependencies/containers/RELEASE_MILESTONE
RELEASE_MILESTONE は、リリース マイルストーン(m88
など)に置き換えます。たとえば、M88 リリースのリストは gs://deeplearning-platform-release/installed-dependencies/containers/m88/
にあります。
TensorFlow Enterprise コンテナ イメージ
TensorFlow Enterprise コンテナ イメージは、Google Cloud 向けに最適化された TensorFlow ディストリビューションを提供します。TensorFlow Enterprise ディストリビューションの特定のバージョンには長期的なバージョン サポートも含まれています。TensorFlow Enterprise の詳細については、TensorFlow Enterprise の概要をご覧ください。
試験運用版イメージ
イメージ ファミリーの表に示されているように、Deep Learning Containers イメージ ファミリーの中には試験運用版のものがあります。試験運用版のイメージはベスト エフォート ベースでサポートされます。また、フレームワークのリリースに合わせてアップデートされない場合があります。
利用可能なすべてのバージョンのリスト
特定のフレームワークや CUDA バージョンが必要な場合は、使用可能なコンテナ イメージの詳細なリストを検索できます。使用可能な Deep Learning Containers イメージをすべて表示するには、任意のターミナルの Google Cloud CLI または Cloud Shell で次のコマンドを使用します。
gcloud container images list --repository="gcr.io/deeplearning-platform-release"
ローカルで使用
Deep Learning Containers はローカルで使用できます。詳細については、ローカルのディープ ラーニング コンテナを使ってみるをご覧ください。
次のステップ
- コンテナ イメージにプリインストールされている内容の詳細について、Deep Learning Containers の概要を確認する。
- 入門ガイドで Deep Learning Containers の使用方法を確認する。ここには、Deep Learning Containers イメージをビルドして push する方法が記載されています。