SeterusnyaPanduan seterusnya
Tensor Parallelism untuk Model Besar
Teknikal
PANDUAN Teknikal
Apabila model terlalu besar untuk dimuatkan pada satu GPU, model dan saluran paip keselarian memisahkan model itu sendiri merentas peranti.
Dengar panduan iniAI-diceritakan · ~2 min
This is what makes training giant language models with hundreds of billions of parameters physically possible.
Model selari membahagikan satu model merentasi berbilang GPU supaya tiada satu peranti perlu memegang semua pemberat. Terdapat dua perisa utama. Keselarian tensor (intra-lapisan) membahagikan matematik di dalam lapisan, seperti memotong pendaraban matriks yang besar merentas GPU yang setiap bahagian mengira output. Keselarian saluran paip (antara lapisan) memberikan lapisan berturut-turut yang berbeza kepada GPU yang berbeza, jadi blok lapisan 1 hidup pada GPU 0, blok 2 pada GPU 1 dan seterusnya, dengan pengaktifan diteruskan seperti barisan pemasangan. Cabaran dengan saluran paip naif ialah 'gelembung': sementara GPU 0 berfungsi pada kumpulan pertama, GPU hiliran terbiar. Saluran paip membahagikan setiap kelompok kepada kelompok mikro supaya semua peringkat kekal sibuk, meningkatkan penggunaan secara mendadak.
Melatih model gaya GPT dengan NVIDIA Megatron-LM, yang membahagikan perhatian setiap lapisan pengubah dan matriks suapan ke hadapan merentas GPU melalui selari tensor.
Menggunakan GPipe untuk meletakkan lapisan berbeza visi atau model bahasa gergasi pada pemecut yang berasingan manakala kumpulan mikro membuatkan mereka sibuk.
Enjin saluran paip DeepSpeed membahagikan model parameter berbilang ratus bilion kepada berperingkat merentas banyak nod.
Menggabungkan selari tensor di dalam pelayan 8-GPU tunggal dengan selari saluran paip yang merangkumi berbilang pelayan untuk melatih model yang terlalu besar untuk satu mesin.
Rangka kerja semakin mengautomasikan masalah sukar untuk memutuskan cara membahagikan model merentas peranti, menggunakan pemprofilan dan carian untuk mengimbangi pengiraan dan komunikasi. Jangkakan penyepaduan yang lebih ketat bagi tensor, saluran paip dan selari data (keselarian 3D), penjadualan kelompok mikro yang lebih bijak untuk hampir menghapuskan gelembung saluran paip, dan perkakasan dengan sambung yang lebih pantas supaya membelah satu lapisan merentas cip menjadi lebih murah dan lebih rutin untuk model yang lebih besar.
Newsletter percuma
Tiga cerita AI yang disahkan setiap pagi hari bekerja, ditulis dalam Bahasa Inggeris biasa. Percuma selamanya, tiada iklan.
Satu e-mel setiap hari bekerja. Berhenti melanggan dalam satu klik. Kami tidak pernah menjual atau berkongsi alamat anda.
Uji diri anda
Maklum balas segera pada setiap jawapan, dan sijil yang boleh dikongsi dengan ID yang boleh disahkan sebaik sahaja anda lulus kursus.
Sokong pendidikan AI percuma. AI Understanding ialah pertubuhan bukan keuntungan 501(c)(3) — tiada iklan, tiada bayaran dinding, selamanya. Buat derma
Apabila model terlalu besar untuk dimuatkan pada satu GPU, model dan saluran paip keselarian memisahkan model itu sendiri merentas peranti. Inilah yang menjadikan latihan model bahasa gergasi dengan ratusan bilion parameter boleh dilakukan secara fizikal.
Keselarian model dan saluran paip membahagikan model itu sendiri merentas peranti, membolehkan latihan rangkaian jauh lebih besar daripada mana-mana GPU tunggal.
Keselarian tensor membahagikan pengiraan dalam satu lapisan, contohnya membelah matriks berat yang besar supaya setiap GPU mengira sebahagian daripada output.
Pada awal langkah saluran paip, peringkat hiliran belum mempunyai input lagi dan terbiar, membuang masa GPU; jurang terbiar ini dipanggil gelembung.
Membahagikan kumpulan kepada kumpulan mikro membolehkan berbilang kelompok mikro menduduki peringkat yang berbeza secara serentak, memastikan semua GPU sibuk dan mengecilkan masa melahu.
Keselarian tensor berkomunikasi dengan kerap untuk menggabungkan semula keputusan matriks separa, jadi ia diletakkan di tempat lebar jalur antara sambungan paling tinggi, di dalam nod di atas NVLink.
Teruskan belajar
Lebih banyak panduan dipilih untuk topik ini
SeterusnyaPanduan seterusnya
Tensor Parallelism untuk Model Besar
Teknikal