Kerangka kerja yang didukung Wilayah AWS Tipe instans yang didukung

Kerangka kerja yang didukung, Wilayah AWS, dan jenis instance

Sebelum menggunakan pust SageMaker aka paralelisme data terdistribusi AI (SMDDP), periksa kerangka kerja ML dan jenis instance yang didukung dan apakah ada cukup kuota di akun Anda dan. AWS Wilayah AWS

Kerangka kerja yang didukung

Tabel berikut menunjukkan kerangka kerja pembelajaran mendalam dan versinya yang didukung SageMaker AI dan SMDDP. Pustaka SMDDP tersedia dalam SageMaker AI Framework Containers, terintegrasi dalam wadah Docker yang didistribusikan oleh perpustakaan SageMaker model paralelism (SMP) v2, atau dapat diunduh sebagai file biner.

catatan

Untuk memeriksa pembaruan terbaru dan catatan rilis pustaka SMDDP, lihat. SageMaker Catatan rilis pustaka paralelisme data AI

PyTorch

PyTorch versi	Versi perpustakaan SMDDP	SageMaker Gambar AI Framework Container pra-instal dengan SMDDP	Gambar SMP Docker pra-instal dengan SMDDP	URL file biner**
v2.3.1	`smdistributed-dataparallel==v2.5.0`	Tidak tersedia	`658645717510.dkr.ecr.<us-west-2>.amazonaws.com/smdistributed-modelparallel:2.4.1-gpu-py311-cu121`	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.4.1/cu121/2024-10-09/smdistributed_dataparallel-2.5.0-cp311-cp311-linux_x86_64.whl`
v2.3.0	`smdistributed-dataparallel==v2.3.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker`	Saat ini tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.3.0/cu121/2024-05-23/smdistributed_dataparallel-2.3.0-cp311-cp311-linux_x86_64.whl`
v2.2.0	`smdistributed-dataparallel==v2.2.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker`	`658645717510.dkr.ecr.<region>.amazonaws.com/smdistributed-modelparallel:2.2.0-gpu-py310-cu121`	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.2.0/cu121/2024-03-04/smdistributed_dataparallel-2.2.0-cp310-cp310-linux_x86_64.whl`
v2.1.0	`smdistributed-dataparallel==v2.1.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker`	`658645717510.dkr.ecr.<region>.amazonaws.com/smdistributed-modelparallel:2.1.2-gpu-py310-cu121`	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.1.0/cu121/2024-02-04/smdistributed_dataparallel-2.1.0-cp310-cp310-linux_x86_64.whl`
v2.0.1	`smdistributed-dataparallel==v2.0.1`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker`	Tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.0.1/cu118/2023-12-07/smdistributed_dataparallel-2.0.2-cp310-cp310-linux_x86_64.whl`
v2.0.0	`smdistributed-dataparallel==v1.8.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.0-gpu-py310-cu118-ubuntu20.04-sagemaker`	Tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.0.0/cu118/2023-03-20/smdistributed_dataparallel-1.8.0-cp310-cp310-linux_x86_64.whl`
v1.13.1	`smdistributed-dataparallel==v1.7.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker`	Tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/1.13.1/cu117/2023-01-09/smdistributed_dataparallel-1.7.0-cp39-cp39-linux_x86_64.whl`
v1.12.1	`smdistributed-dataparallel==v1.6.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:1.12.1-gpu-py38-cu113-ubuntu20.04-sagemaker`	Tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/1.12.1/cu113/2022-12-05/smdistributed_dataparallel-1.6.0-cp38-cp38-linux_x86_64.whl`
v1.12.0	`smdistributed-dataparallel==v1.5.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:1.12.0-gpu-py38-cu113-ubuntu20.04-sagemaker`	Tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/1.12.0/cu113/2022-07-01/smdistributed_dataparallel-1.5.0-cp38-cp38-linux_x86_64.whl`
v1.11.0	`smdistributed-dataparallel==v1.4.1`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:1.11.0-gpu-py38-cu113-ubuntu20.04-sagemaker`	Tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/1.11.0/cu113/2022-04-14/smdistributed_dataparallel-1.4.1-cp38-cp38-linux_x86_64.whl`

** URL file biner adalah untuk menginstal pustaka SMDDP dalam wadah khusus. Untuk informasi selengkapnya, lihat Buat wadah Docker Anda sendiri dengan pustaka paralel data terdistribusi SageMaker AI.

catatan

Pustaka SMDDP tersedia di Wilayah AWS mana SageMaker AI Framework Containers dan gambar Kerangka kerja yang didukung dan Wilayah AWS SMP Docker sedang beroperasi.

catatan

Pustaka SMDDP v1.4.0 dan yang lebih baru berfungsi sebagai backend paralelisme data ter PyTorch distribusi (torch.) (torch.parallel. DistributedDataParallel). Sesuai dengan perubahan, API smdidistribusikan berikut untuk paket ter PyTorch distribusi telah usang.

smdistributed.dataparallel.torch.distributedtidak digunakan lagi. Gunakan paket torch.sebagai gantinya.
smdistributed.dataparallel.torch.parallel.DistributedDataParalleltidak digunakan lagi. Gunakan obor.n n.parallel. DistributedDataParallel API sebagai gantinya.

Jika Anda perlu menggunakan versi pustaka sebelumnya (v1.3.0 atau sebelumnya), lihat dokumentasi paralelisme data ter distribusi SageMaker AI yang diarsipkan dalam dokumentasi SageMaker AI Python SDK.

PyTorch Petir

Pustaka SMDDP tersedia untuk PyTorch Lightning di Kontainer Kerangka SageMaker AI berikut untuk PyTorch dan wadah SMP Docker.

PyTorch Petir v2

PyTorch Versi petir	PyTorch versi	Versi perpustakaan SMDDP	SageMaker Gambar AI Framework Container pra-instal dengan SMDDP	Gambar SMP Docker pra-instal dengan SMDDP	URL file biner**
2.2.5	2.3.0	`smdistributed-dataparallel==v2.3.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.3.0-gpu-py311-cu121-ubuntu20.04-sagemaker`	Saat ini tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.3.0/cu121/2024-05-23/smdistributed_dataparallel-2.3.0-cp311-cp311-linux_x86_64.whl`
2.2.0	2.2.0	`smdistributed-dataparallel==v2.2.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker`	`658645717510.dkr.ecr.<region>.amazonaws.com/smdistributed-modelparallel:2.2.0-gpu-py310-cu121`	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.2.0/cu121/2024-03-04/smdistributed_dataparallel-2.2.0-cp310-cp310-linux_x86_64.whl`
2.1.2	2.1.0	`smdistributed-dataparallel==v2.1.0`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker`	`658645717510.dkr.ecr.<region>.amazonaws.com/smdistributed-modelparallel:2.1.2-gpu-py310-cu121`	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.1.0/cu121/2024-02-04/smdistributed_dataparallel-2.1.0-cp310-cp310-linux_x86_64.whl`
2.1.0	2.0.1	`smdistributed-dataparallel==v2.0.1`	`763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker`	Tidak tersedia	`https://smdataparallel.s3.amazonaws.com/binary/pytorch/2.0.1/cu118/2023-12-07/smdistributed_dataparallel-2.0.2-cp310-cp310-linux_x86_64.whl`

PyTorch Petir v1

PyTorch Versi petir PyTorch versi Versi perpustakaan SMDDP SageMaker Gambar AI Framework Container pra-instal dengan SMDDP URL file biner**

PyTorch Versi petir	PyTorch versi	Versi perpustakaan SMDDP	SageMaker Gambar AI Framework Container pra-instal dengan SMDDP	URL file biner**
1.7.2 1.7.0 1.6.4 1.6.3 1.5.10	1.12.0	`smdistributed-dataparallel==v1.5.0`	763104351884.dkr.ep. `<region>`.amazonaw. com/pytorch-pelatihan: 1.12.0-gpu-py38-cu113-ubuntu20.04-sagemaker	https://smdataparallel.s3.amazonaws.com/binary/pytorch/1.12.0/cu113/2022-07-01/smdistributed_dataparallel-1.5.0-cp38-cp38-linux_x86_64.whl

1.7.2

1.7.0

1.6.4

1.6.3

1.5.10

1.12.0

smdistributed-dataparallel==v1.5.0

763104351884.dkr.ep. <region>.amazonaw. com/pytorch-pelatihan: 1.12.0-gpu-py38-cu113-ubuntu20.04-sagemaker

https://smdataparallel.s3.amazonaws.com/binary/pytorch/1.12.0/cu113/2022-07-01/smdistributed_dataparallel-1.5.0-cp38-cp38-linux_x86_64.whl

** URL file biner adalah untuk menginstal pustaka SMDDP dalam wadah khusus. Untuk informasi selengkapnya, lihat Buat wadah Docker Anda sendiri dengan pustaka paralel data terdistribusi SageMaker AI.

catatan

PyTorch Lightning dan pustaka utilitasnya seperti Lightning Bolts tidak diinstal sebelumnya di PyTorch DLC. Saat Anda membuat SageMaker AI PyTorch ModelTrainer dan mengirimkan permintaan pekerjaan pelatihan di Langkah 2, Anda perlu menyediakan requirements.txt untuk menginstal pytorch-lightning dan lightning-bolts di wadah PyTorch pelatihan SageMaker AI.


# requirements.txt
pytorch-lightning
lightning-bolts

Untuk informasi selengkapnya tentang menentukan direktori sumber untuk menempatkan requirements.txt file bersama dengan skrip pelatihan dan pengiriman pekerjaan, lihat Menggunakan pustaka pihak ketiga dalam dokumentasi Amazon SageMaker AI Python SDK.

Transformer Wajah Memeluk

W AWS adah Pembelajaran Mendalam untuk Hugging Face menggunakan SageMaker Kontainer Pelatihan untuk PyTorch dan TensorFlow sebagai gambar dasarnya. Untuk mencari versi pustaka Hugging Face Transformers dan versi yang dipasangkan PyTorch , lihat Hugging Face Containers terbaru dan Versi Kontainer Hugging Wajah Sebelumnya. TensorFlow

TensorFlow (tidak digunakan lagi)

penting

Pustaka SMDDP menghentikan dukungan untuk TensorFlow dan tidak lagi tersedia di DLC untuk versi lebih baru dari v2.11.0 TensorFlow . Tabel berikut mencantumkan DLC sebelumnya untuk TensorFlow dengan pustaka SMDDP diinstal.

TensorFlow versi	Versi perpustakaan SMDDP
2.9.1, 2.10.1, 2.11.0	`smdistributed-dataparallel==v1.4.1`
2.8.3	`smdistributed-dataparallel==v1.3.0`

Wilayah AWS

Pustaka SMDDP tersedia di semua Wilayah AWS tempat AWS Deep Learning Containers untuk SageMaker AI dan gambar Kerangka kerja yang didukung dan Wilayah AWS SMP Docker beroperasi.

Tipe instans yang didukung

Pustaka SMDDP memerlukan salah satu jenis instance berikut.

Tipe instans
`ml.p3dn.24xlarge`*
`ml.p4d.24xlarge`
`ml.p4de.24xlarge`

Tip

Untuk menjalankan pelatihan terdistribusi dengan benar pada jenis EFA-enabled instans, Anda harus mengaktifkan lalu lintas antar instans dengan mengatur grup keamanan VPC Anda untuk mengizinkan semua lalu lintas masuk dan keluar ke dan dari grup keamanan itu sendiri. Untuk mempelajari cara mengatur aturan grup keamanan, lihat Langkah 1: Mempersiapkan grup EFA-enabled keamanan di Panduan Pengguna Amazon EC2.

penting

* Pustaka SMDDP telah menghentikan dukungan untuk mengoptimalkan operasi komunikasi kolektifnya pada instans P3. Meskipun Anda masih dapat menggunakan AllReduce kolektif yang dioptimalkan SMDDP pada ml.p3dn.24xlarge instans, tidak akan ada dukungan pengembangan lebih lanjut untuk meningkatkan kinerja pada jenis instans ini. Perhatikan bahwa AllGather kolektif yang dioptimalkan SMDDP hanya tersedia untuk instans P4.

Untuk spesifikasi jenis instans, lihat bagian Komput asi Dipercepat di halaman Jenis Instans Amazon EC2. Untuk informasi tentang harga instans, lihat SageMaker Harga Amazon.

Jika Anda menemukan pesan kesalahan yang mirip dengan berikut ini, ikuti petunjuk di Minta peningkatan kuota layanan untuk sumber daya SageMaker AI.


ResourceLimitExceeded: An error occurred (ResourceLimitExceeded) when calling
the CreateTrainingJob operation: The account-level service limit 'ml.p3dn.24xlarge
for training job usage' is 0 Instances, with current utilization of 0 Instances
and a request delta of 1 Instances.
Please contact AWS support to request an increase for this limit.

Awas Javascript dinonaktifkan atau tidak tersedia di browser Anda.

Untuk menggunakan Dokumentasi AWS, Javascript harus diaktifkan. Lihat halaman Bantuan browser Anda untuk petunjuk.

Konvensi Dokumen

Pengantar perpustakaan SMDDP

Pelatihan terdistribusi dengan perpustakaan SMDDP