Container-Image auswählen

Auf dieser Seite können Sie das Container-Image auswählen, das Sie verwenden möchten.

Container-Image-Typ auswählen

Jedes Container-Image stellt eine Python 3-Umgebung bereit und enthält das ausgewählte Data Science-Framework (z. B. PyTorch oder TensorFlow), Conda, den NVIDIA-Stack für GPU-Images (CUDA, cuDNN, NCCL2) und viele weitere hilfreiche Pakete und Tools. In der folgenden Tabelle finden Sie das gewünschte Container-Image.

Die folgende Liste der AI Platform Deep Learning Container-Image-Typen ist nach Framework-Typ organisiert.

Framework Prozessor Container-Image-Namen
Basis GPU gcr.io/deeplearning-platform-release/base-cu100
gcr.io/deeplearning-platform-release/base-cu101
gcr.io/deeplearning-platform-release/base-cu110
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
CPU gcr.io/deeplearning-platform-release/tf2-cpu.2-1
gcr.io/deeplearning-platform-release/tf2-cpu.2-3
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
TensorFlow 2.x GPU gcr.io/deeplearning-platform-release/tf2-gpu
gcr.io/deeplearning-platform-release/tf2-gpu.2-0
gcr.io/deeplearning-platform-release/tf2-gpu.2-2
gcr.io/deeplearning-platform-release/tf2-gpu.2-4
CPU gcr.io/deeplearning-platform-release/tf2-cpu
gcr.io/deeplearning-platform-release/tf2-cpu.2-0
gcr.io/deeplearning-platform-release/tf2-cpu.2-2
gcr.io/deeplearning-platform-release/tf2-cpu.2-4
TensorFlow 1.x GPU gcr.io/deeplearning-platform-release/tf-gpu.1-13
gcr.io/deeplearning-platform-release/tf-gpu.1-14
CPU gcr.io/deeplearning-platform-release/tf-cpu.1-13
gcr.io/deeplearning-platform-release/tf-cpu.1-14
PyTorch GPU gcr.io/deeplearning-platform-release/pytorch-gpu
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
CPU gcr.io/deeplearning-platform-release/pytorch-cpu
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
PyTorch XLA TPU/GPU/CPU (experimentell) 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
R CPU (experimentell) gcr.io/deeplearning-platform-release/r-cpu
gcr.io/deeplearning-platform-release/r-cpu.3-6
Scikit-learn CPU (experimentell) gcr.io/deeplearning-platform-release/sklearn-cpu
gcr.io/deeplearning-platform-release/sklearn-cpu.0-23
XGBoost CPU (experimentell) gcr.io/deeplearning-platform-release/xgboost-cpu
gcr.io/deeplearning-platform-release/xgboost-cpu.1-1

TensorFlow Enterprise-Container-Images

TensorFlow Enterprise-Container-Images bieten Ihnen eine für Google Cloud optimierte Distribution von TensorFlow mit langfristiger Versionsunterstützung. Weitere Informationen zu TensorFlow Enterprise finden Sie in der TensorFlow Enterprise-Übersicht.

Experimentelle Images

Einige Image-Familien für Deep Learning-Container sind experimentell, wie in der Tabelle der Image-Familien angegeben. Experimentelle Images werden auf Best-Effort-Basis unterstützt und werden nicht bei jedem neuen Release des Frameworks automatisch aktualisiert.

Alle verfügbaren Versionen auflisten

Wenn Sie ein bestimmtes Framework oder eine bestimmte CUDA-Version benötigen, können Sie in der vollständigen Liste der verfügbaren Images danach suchen. Verwenden Sie den folgenden Befehl im gcloud-Befehlszeilentool mit Ihrem bevorzugten Terminal oder in Cloud Shell, um alle verfügbaren Deep Learning Container-Images aufzulisten.

gcloud container images list --repository="gcr.io/deeplearning-platform-release"

Lokal verwenden

Deep-Learning-Container können lokal abgerufen und verwendet werden. Weitere Informationen dazu finden Sie unter Erste Schritte mit einem lokalen Deep-Learning-Container.

Nächste Schritte