SıradakiSonraki rehber
Büyük Modeller için Tensör Paralelliği
Teknik
Teknik KILAVUZ
Bir model tek bir GPU'ya sığmayacak kadar büyük olduğunda model ve işlem hattı paralelliği, modeli cihazlar arasında böler.
Bu kılavuzu dinleyinAI tarafından anlatılan · ~2 min
This is what makes training giant language models with hundreds of billions of parameters physically possible.
Model paralelliği, tek bir modeli birden fazla GPU'ya böler, böylece hiçbir cihazın tüm ağırlığı taşımasına gerek kalmaz. İki ana lezzet var. Tensör (katman içi) paralelliği, her biri çıktının bir kısmını hesaplayan GPU'lar arasında büyük bir matris çarpımının kesilmesi gibi, matematiği bir katman içinde böler. İşlem hattı (katmanlar arası) paralelliği, farklı GPU'lara farklı ardışık katmanlar atar, böylece katman bloğu 1 GPU 0'da, blok 2 GPU 1'de yaşar ve aktivasyonlar bir montaj hattı gibi ileriye aktarılarak böyle devam eder. Saf ardışık düzen oluşturmanın zorluğu 'balon'dur: GPU 0 ilk partide çalışırken, aşağı yöndeki GPU'lar boşta kalır. İşlem hattı, her partiyi mikro partilere bölerek tüm aşamaların meşgul kalmasını sağlar ve kullanımı önemli ölçüde artırır.
Tensör paralelliği yoluyla her transformatör katmanının dikkatini ve ileri besleme matrislerini GPU'lar arasında bölen NVIDIA Megatron-LM ile GPT tarzı modelleri eğitme.
Mikro gruplama onları meşgul ederken dev bir vizyonun veya dil modelinin farklı katmanlarını ayrı hızlandırıcılara yerleştirmek için GPipe'ı kullanmak.
DeepSpeed'in yüz milyarlarca parametreli bir modeli birçok düğümdeki aşamalara bölen boru hattı motoru.
Tek bir 8 GPU'lu sunucu içindeki tensör paralelliğini, tek bir makine için çok büyük bir modeli eğitmek üzere birden fazla sunucuya yayılan boru hattı paralelliğiyle birleştirmek.
Çerçeveler, bilgi işlem ve iletişimi dengelemek için profil oluşturma ve aramayı kullanarak, bir modelin cihazlar arasında nasıl bölümleneceğine karar verme gibi zor bir sorunu giderek daha fazla otomatik hale getiriyor. Tensör, boru hattı ve veri paralelliğinin (3D paralellik) daha sıkı entegrasyonunu, boru hattı kabarcıklarını neredeyse ortadan kaldıracak daha akıllı mikro toplu planlamayı ve daha hızlı ara bağlantılara sahip donanımın, böylece tek bir katmanın çipler arasında bölünmesinin giderek daha büyük modeller için daha ucuz ve daha rutin hale gelmesini bekleyin.
Ücretsiz bülten
Hafta içi her sabah sade İngilizce yazılmış, doğrulanmış üç yapay zeka hikayesi. Sonsuza kadar ücretsiz, reklam yok.
Hafta içi her gün bir e-posta. Tek tıklamayla aboneliğinizi iptal edin. Adresinizi asla satmayız veya paylaşmayız.
Kendinizi test edin
Her yanıtla ilgili anında geri bildirim ve bir kursu geçtiğinizde doğrulanabilir bir kimliğe sahip paylaşılabilir bir sertifika.
Ücretsiz AI eğitimini destekleyin. AI Understanding, bir 501(c)(3) kâr amacı gütmeyen kuruluştur; hiçbir zaman reklam yoktur, ödeme duvarı yoktur. Bağış yap
Bir model tek bir GPU'ya sığmayacak kadar büyük olduğunda model ve işlem hattı paralelliği, modeli cihazlar arasında böler. Yüz milyarlarca parametreye sahip dev dil modellerinin eğitilmesini fiziksel olarak mümkün kılan şey budur.
Model ve boru hattı paralelliği, modelin kendisini cihazlar arasında bölerek herhangi bir GPU'nun taşıyamayacağı kadar büyük ağların eğitilmesine olanak tanır.
Tensör paralelliği hesaplamayı tek bir katmana böler; örneğin büyük bir ağırlık matrisini her GPU'nun çıktının bir kısmını hesaplaması için böler.
Boru hattı adımının başlarında, alt aşamaların henüz girişi yoktur ve boşta bekleyerek GPU zamanını boşa harcar; bu boş boşluğa balon denir.
Bir grubun mikro gruplara bölünmesi, birden fazla mikro grubun aynı anda farklı aşamaları işgal etmesine olanak tanıyarak tüm GPU'ları meşgul eder ve boşta kalma süresini kısaltır.
Tensör paralelliği, kısmi matris sonuçlarını yeniden birleştirmek için sıklıkla iletişim kurar; bu nedenle, ara bağlantı bant genişliğinin en yüksek olduğu yere, NVLink üzerinden bir düğümün içine yerleştirilir.
Öğrenmeye devam et
Bu konu için daha fazla rehber seçildi
SıradakiSonraki rehber
Büyük Modeller için Tensör Paralelliği
Teknik