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 data science (seperti PyTorch atau TensorFlow), Conda, stack NVIDIA untuk image GPU (CUDA, cuDNN, NCCL2), serta banyak paket dan 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 |
|---|---|---|---|---|---|
| Dasar-CU121 | CUDA 12.1 (Python 3.10) | GPU (CUDA 12.1) | 28 Feb 2024 | 28 Feb 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310 |
| Dasar-CU118 | CUDA 11.8 (Python 3.10) | GPU (CUDA 11.8) | 1 Juli 2024 | 1 Juli 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-gpu.py310 |
| Base-cu113 (Python 3.10) | CUDA 11.3 | GPU (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 | GPU (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 | GPU (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 | GPU (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.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 | GPU (CUDA 12.1) | 14 Nov 2024 | 14 Nov 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-15.py310 |
| 2.14 (Python 3.10) | 2.14.0 | Khusus CPU | 26 September 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 | GPU (CUDA 11.8) | 26 September 2024 | 26 September 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 5 Juli 2024 | 5 Juli 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 30 Jun 2024 | 30 Jun 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.3) | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.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 | GPU (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 | GPU (CUDA 11.3) | 15 Nov 2023 | 15 Nov 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.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 | GPU (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 | GPU (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 | GPU (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 | GPU (CUDA 11.3) | 10 Agustus 2023 | 10 Agu 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.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 | GPU (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 | GPU (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.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-gpu.2-2.py310 |
| 2.1 (Python 3.10) | 2.1.0 | CUDA 12.1 | 4 Oktober 2024 | 4 Oktober 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.2-1.py310 |
| 2.0 (Python 3.10) | 2.0.0 | CUDA 11.8 | 15 Maret 2024 | 15 Maret 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.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-gpu.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-gpu.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 |
Gambar container Wajah Memeluk
Bagian ini mencantumkan gambar container Wajah Membungkuk yang tersedia.
Penampung Inferensi Pembuatan Teks
| Versi framework ML | Versi patch saat ini | Akselerator yang didukung | Akhir patch dan tanggal dukungan | Akhir tanggal ketersediaan | Nama kelompok image |
|---|---|---|---|---|---|
| PyTorch 2.1 (Python 3.10) | 2.1.1 | CUDA 12.1 | 4 Oktober 2024 | 4 Oktober 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310 |
Kelompok gambar eksperimental
Kelompok gambar yang tidak tercantum di sini masih bersifat eksperimental. Kelompok gambar eksperimental didukung atas dasar upaya terbaik, dan tidak menerima pembaruan pada setiap rilis framework baru.
Mencantumkan semua versi yang tersedia
Jika Anda memerlukan framework atau versi CUDA tertentu, telusuri daftar lengkap image penampung yang tersedia. Untuk menampilkan 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 Mulai menggunakan penampung deep learning lokal.
Langkah selanjutnya
- Baca ringkasan Deep Learning Containers untuk mempelajari lebih lanjut apa saja yang sudah diinstal sebelumnya pada image container.
- Mulai Deep Learning Containers dengan mengikuti Panduan cara kerja, yang memberikan petunjuk tentang cara membangun dan mengirim image container deep learning.