Halaman ini membantu Anda memilih image container yang akan digunakan.
Memilih jenis image container
Deep Learning Containers mendukung setiap versi framework berdasarkan jadwal untuk meminimalkan kerentanan keamanan. Tinjau kebijakan dukungan framework Deep Learning Containers untuk memahami implikasi dari tanggal akhir dukungan dan akhir ketersediaan.
Setiap image container menyediakan lingkungan Python 3 dan menyertakan framework data science yang dipilih (seperti PyTorch atau TensorFlow), Conda, stack NVIDIA untuk image GPU (CUDA, cuDNN, NCCL2), dan banyak paket dan alat pendukung lainnya. Untuk menemukan image penampung yang sesuai, lihat tabel di bawah.
Versi Base
| Versi framework ML | Versi patch saat ini | Akselerator yang didukung | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|
| Base-cu124 | CUDA 12.4 (Python 3.10) | CUDA 12.4 | 1 April 2025 | 1 April 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu124.py310 |
| Base-cu123 | CUDA 12.3 (Python 3.10) | CUDA 12.3 | 1 Jan 2025 | 1 Januari 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu123.py310 |
| Base-cu122 | CUDA 12.2 (Python 3.10) | CUDA 12.2 | 8 Jan 2025 | 8 Januari 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu122.py310 |
| Base-cu121 | CUDA 12.1 (Python 3.10) | CUDA 12.1 | 28 Feb 2024 | 28 Februari 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310 |
| Base-cu118 | CUDA 11.8 (Python 3.10) | CUDA 11.8 | 1 Juli 2024 | 1 Juli 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu118.py310 |
| Base-cu113 (Python 3.10) | CUDA 11.3 | CUDA 11.3 | 1 Jan 2024 | 1 Jan 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py310 |
| Base-cu113 (Python 3.7) | CUDA 11.3 | CUDA 11.3 | 1 Sep 2023 | 1 Sep 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py37 |
| Base-cu110 (Python 3.10 / Debian 11) | CUDA 11.0 | CUDA 11.0 | 1 Jan 2024 | 1 Jan 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py310 |
| Base-cu110 (Python 3.7) | CUDA 11.0 | CUDA 11.0 | 1 Sep 2023 | 1 Sep 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37 |
Versi TensorFlow
| Versi framework ML | Versi patch saat ini | Akselerator yang didukung | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|
| 2.17 (Python 3.10) | 2.17.0 | Khusus CPU | 11 Juli 2025 | 11 Juli 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-17.py310 |
| 2.17 (Python 3.10) | 2.17.0 | CUDA 12.3 | 11 Juli 2025 | 11 Juli 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-17.py310 |
| 2.16 (Python 3.10) | 2.16.2 | Khusus CPU | 28 Jun 2025 | 28 Juni 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-16.py310 |
| 2.16 (Python 3.10) | 2.16.2 | CUDA 12.3 | 28 Jun 2025 | 28 Juni 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-16.py310 |
| 2.15 (Python 3.10) | 2.15.0 | Khusus CPU | 14 November 2024 | 14 November 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-15.py310 |
| 2.15 (Python 3.10) | 2.15.0 | CUDA 12.1 | 14 November 2024 | 14 November 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu121.2-15.py310 |
| 2.14 (Python 3.10) | 2.14.0 | Khusus CPU | 26 Sep 2024 | 26 September 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-14.py310 |
| 2.14 (Python 3.10) | 2.14.0 | CUDA 11.8 | 26 Sep 2024 | 26 September 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-14.py310 |
| 2.13 (Python 3.10) | 2.13.0 | Khusus CPU | 5 Juli 2024 | 5 Juli 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-13.py310 |
| 2.13 (Python 3.10) | 2.13.0 | CUDA 11.8 | 5 Juli 2024 | 5 Juli 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-13.py310 |
| 2.12 (Python 3.10) | 2.12.0 | Khusus CPU | 30 Jun 2024 | 30 Jun 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-12.py310 |
| 2.12 (Python 3.10) | 2.12.0 | CUDA 11.8 | 30 Jun 2024 | 30 Jun 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-12.py310 |
| 2.11 (Python 3.10) | 2.11.0 | Khusus CPU | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py310 |
| 2.11 (Python 3.10) | 2.11.0 | CUDA 11.3 | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-11.py310 |
| 2.11 | 2.11.0 | Khusus CPU | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py37 |
| 2.11 | 2.11.0 | CUDA 11.3 | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-11.py37 |
| 2.10 (Python 3.10) | 2.10.1 | Khusus CPU | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py310 |
| 2.10 (Python 3.10) | 2.10.1 | CUDA 11.3 | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-10.py310 |
| 2.10 | 2.10.1 | Khusus CPU | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py37 |
| 2.10 | 2.10.1 | CUDA 11.3 | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-10.py37 |
| 2.9 | 2.9.3 | Khusus CPU | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-9.py37 |
| 2.9 | 2.9.3 | CUDA 11.3 | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-9.py37 |
| 2.8 | 2.8.4 | Khusus CPU | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-8.py37 |
| 2.8 | 2.8.4 | CUDA 11.3 | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-8.py37 |
| 2.6 (Python 3.9) | 2.6.5 | Khusus CPU | 10 Agustus 2023 | 10 Agu 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-6.py39 |
| 2.6 (Python 3.9) | 2.6.5 | CUDA 11.3 | 10 Agustus 2023 | 10 Agu 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-6.py39 |
| 2.6 (Python 3.7) | 2.6.5 | Khusus CPU | 1 Sep 2023 | 1 Sep 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-6.py37 |
| 2.6 (Python 3.7) | 2.6.5 | CUDA 11.3 | 1 Sep 2023 | 1 Sep 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37 |
| 2.3 | 2.3.4 | Khusus CPU | 1 Sep 2023 | 1 Sep 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-3.py37 |
| 2.3 | 2.3.4 | CUDA 11.3 | 1 Sep 2023 | 1 Sep 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37 |
Versi PyTorch
| Versi framework ML | Versi patch saat ini | Akselerator yang didukung | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|
| 2.4 (Python 3.10) | 2.4.0 | CUDA 12.4 | 24 Juli 2025 | 24 Juli 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu124.2-4.py310 |
| 2.3 (Python 3.10) | 2.3.0 | CUDA 12.1 | 24 April 2025 | 24 April 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu121.2-3.py310 |
| 2.2 (Python 3.10) | 2.2.0 | CUDA 12.1 | 30 Jan 2025 | 30 Januari 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu121.2-2.py310 |
| 2.1 (Python 3.10) | 2.1.0 | CUDA 12.1 | 4 Okt 2024 | 4 Oktober 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu121.2-1.py310 |
| 2.0 (Python 3.10) | 2.0.0 | CUDA 11.8 | 15 Mar 2024 | 15 Mar 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu118.2-0.py310 |
| 1.13 (Python 3.10) | 1.13.1 | CUDA 11.8 | 8 Des 2023 | 8 Des 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu118.1-13.py310 |
| 1.13 | 1.13.1 | CUDA 11.8 | 8 Des 2023 | 8 Des 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-13.py37 |
| 1.12 (Python 3.10) | 1.12.1 | CUDA 11.3 | 8 Des 2023 | 8 Des 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu113.1-12.py310 |
| 1.12 | 1.12.1 | CUDA 11.3 | 1 Sep 2023 | 1 Sep 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37 |
Image container Hugging Face
Bagian ini mencantumkan Deep Learning Containers Hugging Face yang tersedia.
Penampung Inferensi Pembuatan Teks
| Versi framework | Versi patch saat ini | Akselerator yang didukung | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|
| TGI 2.4 | 2.4.0 | CUDA 12.4 | 5 Desember 2025 | 5 Desember 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu124.2-4.ubuntu2204.py311 |
| TGI 2.3 | 2.3.1 | CUDA 12.4 | 9 Oktober 2025 | 9 Oktober 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu124.2-3.ubuntu2204.py311 |
| TGI 2.2 | 2.2.0 | CUDA 12.1 | 23 Juli 2025 | 23 Juli 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-2.ubuntu2204.py310 |
| TGI 2.1 | 2.1.1 | CUDA 12.1 | 27 Juni 2025 | 27 Juni 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-1.ubuntu2204.py310 |
| TGI 2.0 | 2.0.1 | CUDA 12.1 | 25 April 2025 | 25 April 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-0.ubuntu2204.py310 |
| TGI 1.4 | 1.4.4 | CUDA 12.1 | 13 Maret 2025 | 13 Maret 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310 |
Penampung Inferensi Embedding Teks
| Versi framework | Versi patch saat ini | Akselerator yang didukung | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|
| TEI 1.4 | 1.4.0 | CUDA 12.2 | 27 Agustus 2025 | 27 Agustus 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-4.ubuntu2204 |
| TEI 1.4 | 1.4.0 | CPU | 27 Agustus 2025 | 27 Agustus 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-4 |
| TEI 1.2 | 1.2.2 | CUDA 12.2 | 2 Mei 2025 | 2 Mei 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-2.ubuntu2204 |
| TEI 1.2 | 1.2.2 | CPU | 2 Mei 2025 | 2 Mei 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-2 |
Penampung Inferens PyTorch
| Versi framework | Versi patch saat ini | Akselerator yang didukung | Versi Transformers | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|---|
| PyTorch 2.3 | 2.3.1 | CUDA 12.1 | 4.46.1 | 7 November 2025 | 7 November 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-3.transformers.4-46.ubuntu2204.py311 |
| PyTorch 2.3 | 2.3.1 | CPU | 4.46.1 | 12 Nov 2025 | 12 Nov 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-3.transformers.4-46.ubuntu2204.py311 |
| PyTorch 2.2 | 2.2.2 | CUDA 12.1 | 4.41.1 | 12 Juni 2025 | 12 Juni 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-41.ubuntu2204.py311 |
| PyTorch 2.2 | 2.2.2 | CPU | 4.41.1 | 7 Juni 2025 | 7 Juni 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-41.ubuntu2204.py311 |
| PyTorch 2.2 | 2.2.2 | CUDA 12.1 | 4.44.0 | 21 Agustus 2025 | 21 Agustus 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-44.ubuntu2204.py311 |
| PyTorch 2.2 | 2.2.2 | CPU | 4.44.0 | 21 Agustus 2025 | 21 Agustus 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311 |
Container pelatihan PyTorch
| Versi framework | Versi patch saat ini | Akselerator yang didukung | Versi Transformers | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|---|
| PyTorch 2.3 | 2.3.0 | CUDA 12.1 | 4.42.3 | 21 Agustus 2025 | 21 Agustus 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.2-3.transformers.4-42.ubuntu2204.py310 |
Image container Model Garden
Bagian ini mencantumkan image penampung Model Garden yang tersedia.
Penampung inferensi vLLM
| Versi framework | Versi patch saat ini | Akselerator yang didukung | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|
| PyTorch 2.4 | 2.4.0 | CUDA 12.1 | 21 Agustus 2025 | 21 Agustus 2026 | us-docker.pkg.dev/deeplearning-platform-release/vertex-model-garden/vllm-inference.cu121.0-5.ubuntu2204.py310 |
Kelompok image eksperimental
Kelompok image yang tidak tercantum di sini bersifat eksperimental. Grup image eksperimental didukung atas dasar upaya terbaik, dan tidak menerima pembaruan pada setiap rilis baru framework.
Mencantumkan semua versi yang tersedia
Jika Anda memerlukan framework atau versi CUDA tertentu, telusuri daftar lengkap image container yang tersedia. Untuk mencantumkan semua image Deep Learning Containers yang tersedia, gunakan perintah berikut di Google Cloud CLI dengan terminal pilihan Anda atau di Cloud Shell.
gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"
Menggunakan secara lokal
Deep Learning Containers dapat diambil dan digunakan secara lokal. Untuk melakukannya, lihat Memulai container deep learning lokal.
Langkah selanjutnya
- Baca ringkasan Deep Learning Containers untuk mempelajari lebih lanjut apa yang telah diinstal sebelumnya di image container.
- Mulai menggunakan Deep Learning Containers dengan mempelajari Panduan cara kerja, yang memberikan petunjuk tentang cara mem-build dan mendorong image container deep learning.