Kompatibilitas dengan SMDDP pustaka yang dioptimalkan untuk AWS infrastruktur

Anda dapat menggunakan pustaka paralelisme SageMaker model v2 (SMPv2) bersama dengan pustaka paralelisme data SageMaker terdistribusi (SMDDP) yang menawarkan operasi komunikasi kolektif yang dioptimalkan untuk AllGather AWS Infrastruktur. Dalam pelatihan terdistribusi, operasi komunikasi kolektif dirancang untuk menyinkronkan banyak GPU pekerja dan bertukar informasi di antara mereka. AllGatheradalah salah satu operasi komunikasi kolektif inti yang biasanya digunakan dalam paralelisme data sharded. Untuk mempelajari lebih lanjut tentang SMDDP AllGather operasi, lihat Operasi kolektif SMDDP AllGather Mengoptimalkan operasi komunikasi kolektif semacam itu akan secara langsung berkontribusi pada end-to-end pelatihan yang lebih cepat tanpa efek samping pada konvergensi.

SMDDPPustaka terintegrasi secara native dengan PyTorch melalui lapisan grup proses. Untuk menggunakan SMDDP perpustakaan, Anda hanya perlu menambahkan dua baris kode ke skrip pelatihan Anda. Ini mendukung kerangka kerja pelatihan seperti SageMaker Model Parallelism Library,, PyTorch FSDP dan. DeepSpeed

Untuk mengaktifkan SMDDP dan menggunakan AllGather operasinya, Anda perlu menambahkan dua baris kode ke skrip pelatihan Anda sebagai bagian dariLangkah 1: Sesuaikan skrip pelatihan PyTorch FSDP Anda. Perhatikan bahwa Anda perlu menginisialisasi PyTorch Distributed dengan SMDDP backend terlebih dahulu, dan kemudian menjalankan inisialisasi. SMP

import torch.distributed as dist

# Initialize with SMDDP
import smdistributed.dataparallel.torch.torch_smddp
dist.init_process_group(backend="smddp") # Replacing "nccl"

 # Initialize with SMP
import torch.sagemaker as tsm
tsm.init()

SageMaker Wadah Kerangka untuk PyTorch (lihat juga Kerangka kerja yang didukung dan Wilayah AWS oleh SMP v2 dan Kerangka kerja yang didukung, Wilayah AWS, dan tipe instance oleh SMDDP perpustakaan) sudah dikemas sebelumnya dengan SMP biner dan biner. SMDDP Untuk mempelajari lebih lanjut tentang SMDDP perpustakaan, lihatJalankan pelatihan terdistribusi dengan SageMaker pustaka paralelisme data terdistribusi.