コンテナ イメージを選択する

このページでは、使用するコンテナ イメージの選択に役立つ情報を提供します。

コンテナ イメージのタイプを選択する

各コンテナ イメージは 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/tf2-gpu.2-1
gcr.io/deeplearning-platform-release/tf2-gpu.2-3
gcr.io/deeplearning-platform-release/tf2-gpu.2-6
gcr.io/deeplearning-platform-release/tf2-gpu.2-7
CPU gcr.io/deeplearning-platform-release/tf2-cpu.2-1
gcr.io/deeplearning-platform-release/tf2-cpu.2-3
gcr.io/deeplearning-platform-release/tf2-cpu.2-6
gcr.io/deeplearning-platform-release/tf2-cpu.2-7
TensorFlow Enterprise 1.x GPU gcr.io/deeplearning-platform-release/tf-gpu
gcr.io/deeplearning-platform-release/tf-gpu.1-15
CPU gcr.io/deeplearning-platform-release/tf-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-10
PyTorch XLA TPU / GPU / CPU(試験運用版) gcr.io/deeplearning-platform-release/pytorch-xla
gcr.io/deeplearning-platform-release/pytorch-xla.1-10
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
非推奨の Deep Learning Container イメージ 送受信 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/tf2-gpu.2-5
gcr.io/deeplearning-platform-release/tf2-cpu.2-5
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-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

TensorFlow Enterprise コンテナ イメージ

TensorFlow Enterprise コンテナ イメージは、Google Cloud 向けに最適化された TensorFlow ディストリビューションを提供します。TensorFlow Enterprise ディストリビューションの特定のバージョンには長期的なバージョン サポートも含まれています。TensorFlow Enterprise の詳細については、TensorFlow Enterprise の概要をご覧ください。

試験運用版イメージ

イメージ ファミリーの表に示されているように、Deep Learning Containers イメージ ファミリーの中には試験運用版のものがあります。試験運用版のイメージはベスト エフォート ベースでサポートされます。また、フレームワークのリリースに合わせてアップデートされない場合があります。

利用可能なすべてのバージョンのリスト

特定のフレームワークや CUDA バージョンが必要な場合は、使用可能なコンテナ イメージの詳細なリストを検索できます。使用可能なすべての Deep Learning Containers イメージを一覧表示するには、任意のターミナルで gcloud コマンドライン ツールを使用するか、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 する方法が記載されています。