Memilih image container

Halaman ini membantu Anda memilih image container yang akan digunakan.

Pilih 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 mencakup framework ilmu data yang dipilih (seperti PyTorch atau TensorFlow), Conda, stack NVIDIA untuk image GPU (CUDA, cuDNN, NCCL2), dan banyak paket serta alat pendukung lainnya. Untuk menemukan image container 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 Jan 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 Jan 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 Jun 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 Jun 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 Nov 2024 14 Nov 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 Nov 2024 14 Nov 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 Sep 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 Sep 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 Jan 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 Okt 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 Model Garden

Bagian ini mencantumkan image container Model Garden yang tersedia.

Container 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 Agu 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. Kelompok 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 ditarik dan digunakan secara lokal. Untuk melakukannya, lihat Mulai menggunakan Deep Learning Container lokal.

Melihat kemunculan kerentanan

Untuk melihat kemunculan kerentanan pada image Deep Learning Containers, lihat Melihat kemunculan menggunakan gcloud.

Langkah berikutnya

  • Baca ringkasan Deep Learning Containers untuk mempelajari lebih lanjut apa yang telah diinstal sebelumnya pada image container.
  • Mulai gunakan Deep Learning Containers dengan mempelajari Panduan cara kerja, yang memberikan petunjuk tentang cara membuat dan mengirimkan image container deep learning.