SıradakiSonraki rehber
Adam ve Uyarlanabilir Optimize Ediciler
Teknik
Teknik KILAVUZ
Yoğun eğitim muhasebesini (optimizer durumları, eğimler, bazen ağırlıklar) kıt GPU belleği yerine CPU RAM'e veya NVMe SSD'lere park eden, bellek tasarrufu sağlayan bir numara.
Bu kılavuzu dinleyinAI tarafından anlatılan · ~2 min
Bu, insanların GPU'nun belleğinden çok daha büyük modelleri eğitmesine olanak tanıyor.
Bir sinir ağını Adam gibi bir optimize ediciyle eğittiğinizde, her parametre ekstra yük taşır: iki çalışan istatistik (momentum ve varyans), artı ağırlığın tam hassasiyetli bir kopyası ve eğimi. Karma duyarlıklı eğitimde bu, parametre başına kabaca 16 bayta ulaşabilir ve ağırlığın kendisi için 2 baytı gölgede bırakır. Boşaltma bu bagajı GPU'dan uzaklaştırır. CPU boşaltma, optimize edici durumlarını PCIe veri yolu üzerinden sıradan sistem RAM'ine aktarırken NVMe boşaltma, bunları hızlı katı hal disklerine kadar iter. DeepSpeed'in Zero-Infinity ve Zero-Offload özellikleriyle popüler hale gelen bu teknik, ham hızı kapasiteyle takas ederek tek bir GPU'nun veya küçük kümenin milyarlarca parametreli modellere ince ayar yapmasına olanak tanıyor.
DeepSpeed’s ZeRO-Offload tutorial trains a ten-billion-parameter GPT-2 configuration on a 32 GB V100, with CPU optimizer offloading and activation checkpointing.
A research team evaluates ZeRO-Infinity with CPU and NVMe tiers for a model that exceeds GPU-only capacity.
A fine-tuning team configures CPU optimizer offloading and measures host memory use and update time.
A team compares larger GPUs with offloading, accounting for CPU, storage, and transfer costs.
Modeller GPU belleğini aşmaya devam ettikçe, katmanlı boşaltma egzotik olmaktan çıkıp standart hale geliyor. CPU-GPU sınırını bulanıklaştıran NVLink-C2C ve CXL bellek havuzları gibi daha hızlı ara bağlantıların yanı sıra hangi durumların önceden getirileceğini tahmin eden daha akıllı zamanlayıcılarla daha sıkı entegrasyon bekleyebilirsiniz. Grace Hopper gibi birleşik bellek mimarileri PCIe cezasını azaltıyor ve çerçeveler, hobi meraklılarının büyük modellere mütevazı donanımlarda ince ayar yapabilmesi için çok katmanlı boşaltmayı neredeyse şeffaf hale getirmeye çalışıyor.
Ü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
Yoğun eğitim muhasebesini (optimizer durumları, eğimler, bazen ağırlıklar) kıt GPU belleği yerine CPU RAM'e veya NVMe SSD'lere park eden, bellek tasarrufu sağlayan bir numara. İnsanların GPU belleğinin izin verebileceğinden çok daha büyük modelleri eğitmesine olanak tanır.
Boşaltma, ağır optimize edici durumlarını GPU'dan CPU RAM'e veya NVMe'ye aktararak GPU belleğini serbest bırakır, böylece daha büyük modeller aynı donanım üzerinde eğitilebilir.
Each FP32 value uses four bytes: two moments plus one master weight total twelve. A sixteen-byte mixed-precision estimate includes other tensors as well.
DeepSpeed'in Sıfır Boşaltma ve Sıfır Sonsuzluk, boşaltma optimize edici durumlarını CPU ve NVMe'ye uygun ölçekte tanıttı ve yaygınlaştırdı.
Durumları GPU dışına taşımak, verileri PCIe üzerinden veya GPU üzerindeki bellekten daha yavaş olan NVMe'ye aktarmak anlamına gelir; dolayısıyla işlemle çakışmadığı sürece aktarım hızı düşer.
Zamanlayıcılar, GPU hala bilgi işlem yaparken, gerekli bölümleri PCIe üzerinden önceden getirir ve gecikmeyi maskelemek için iletişimi hesaplamayla örtüştürür.
Öğrenmeye devam et
Bu konu için daha fazla rehber seçildi
SıradakiSonraki rehber
Adam ve Uyarlanabilir Optimize Ediciler
Teknik