Kerangka kerja dan AWS Wilayah yang didukung - Amazon SageMaker AI

Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Kerangka kerja dan AWS Wilayah yang didukung

Sebelum menggunakan pemuat data penyaringan SageMaker cerdas, periksa apakah kerangka kerja pilihan Anda didukung, apakah jenis instans tersedia di AWS akun Anda, dan apakah AWS akun Anda berada di salah satu AWS Wilayah yang didukung.

catatan

SageMaker smart sifting mendukung pelatihan PyTorch model dengan paralelisme data tradisional dan paralelisme data terdistribusi, yang membuat replika model di semua pekerja GPU dan menggunakan operasi. AllReduce Ini tidak bekerja dengan teknik paralelisme model, termasuk paralelisme data sharded. Karena penyaringan SageMaker cerdas berfungsi untuk pekerjaan paralelisme data, pastikan model yang Anda latih cocok di setiap memori GPU.

Kerangka Kerja yang Didukung

SageMaker smart sifting mendukung kerangka pembelajaran mendalam berikut dan tersedia melalui Deep Learning AWS Containers.

Topik

PyTorch

Kerangka Kerja Versi kerangka URI Wadah Pembelajaran Mendalam
PyTorch 2.1.0

763104351884.dkr.ecr. region.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker

Untuk informasi selengkapnya tentang container yang sudah dibuat sebelumnya, lihat SageMaker AI Framework Container di GitHub repositori AWS Deep Learning Containers.

Wilayah AWS

Wadah yang dikemas dengan perpustakaan penyaringan SageMaker pintar tersedia di Wilayah AWS tempat AWS Deep Learning Containers berada dalam layanan.

Tipe instans

Anda dapat menggunakan penyaringan SageMaker cerdas untuk pekerjaan PyTorch pelatihan apa pun pada jenis instans apa pun. Kami menyarankan Anda menggunakan instans P4d, P4de, atau P5.