Kerangka kerja dan AWS Wilayah yang didukung - Amazon SageMaker

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 GPU pekerja 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 kerja Wadah Pembelajaran Mendalam URI
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 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. Sebaiknya gunakan instans P4d, P4de, atau P5.