Nhungamiro yehunyanzvi

Optimizer State Offloading kuCPU uye NVMe

Chirevo chekuchengetedza ndangariro chinomisa kuchengetwa kwebhuku kunorema kwekudzidziswa (optimizer states, gradients, dzimwe nguva uremu) muCPU RAM kana paNVMe SSDs pane kushomeka GPU memory.

Teerera kugwaro iriAI-yakarondedzerwa · ~2 min

Pa peji ino2 min verenga
  1. Pfupiso
  2. Kudzika Kwakadzika
  3. Kuitwa muNyika yeChokwadi
  4. Ramangwana reOptimizer State Offloading kuCPU uye NVMe
  5. Kuzvishandisa
  6. Ramba Uchiongorora
  7. Mibvunzo inowanzo bvunzwa

Pfupiso

It lets people train far larger models than their GPU's memory would otherwise allow.

Kudzika Kwakadzika

Paunodzidzisa neural network ine optimizer senge Adamu, imwe neimwe parameter inotakura mukwende wakawedzerwa: maviri anomhanya manhamba (kumhanya uye musiyano), pamwe neiyo yakazara-chaiyo kopi yehuremu, pamwe negradient yayo. Mukudzidziswa kwakasanganiswa-chaizvo izvi zvinogona kuita zvinosvika gumi nematanhatu mabhayiti paparamita, kudikitira mabhayiti maviri ehuremu hwacho. Kuburitsa kutakura mutoro iwoyo kubva paGPU. CPU inoburitsa hova optimizer inopinda mune yakajairwa system RAM pamusoro pePCIe bhazi, ukuwo NVMe kuburitsa ichivasundidzira nzira yese kudzika kukurumidza solid-state disks. Yakasimudzirwa neDeepSpeed's ZeRO-Infinity uye ZeRO-Offload, nzira iyi inotengeserana nekumhanya kwehuwandu, ichirega imwe GPU kana diki diki-tune modhi ine mabhiriyoni emaparamita.

Kuitwa muNyika yeChokwadi

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.

Ramangwana reOptimizer State Offloading kuCPU uye NVMe

Sezvo mamodheru achiramba achikura ndangariro yeGPU, kurodha kwakapetwa kuri kuita kwakajairwa kwete kwekunze. Tarisira kubatanidzwa kwakasimba nekukurumidza kusanganisa seNVLink-C2C neCXL ndangariro madziva anodzima muganho weCPU-GPU, pamwe neanoronga kuronga anofanotaura kuti ndezvipi zvinofanirwa kufanotorwa. Akabatana-memory architecture akadai saGrace Hopper anoderedza chirango chePCIe, uye masisitimu ari kusundira akananga kugadzira akawanda-tier kurodha zvakada kujeka kuitira kuti hobbyists vagone kukwenenzvera mamodheru pane zvine mwero Hardware.

Kuzvishandisa

Nei zvichikosha

  • Mutengo uye bhajeti. Zvisarudzo zvezvivakwa zvinotyaira kuita uye mutengo wekushandisa kwemakore.
  • Sarudzo dzakajeka. Dzidzo yehunyanzvi inobatsira zvikwata kusarudza murwi wakakodzera, kwete iwo mutsva chete.
  • Kudzora kwemhando yepamusoro. Sarudzo dzeinjiniya dziri nani dzinoderedza zviitiko zvekuvimbika mukugadzira.

Chenjera

  • Kugadzirisa imwe bhenji kunogona kuvanza yakafara system kushaya simba.
  • Infrastructure uye mari yekugadzirisa inowanzotarisirwa pasi.
  • Chengetedzo uye kucherechedzwa mapundu anogona kukura sezvo masisitimu anowedzera kuoma.

Matanho anotevera

  1. Tsanangura latency, mhando, uye mutengo zvinangwa usati waitwa.
  2. Benchmark pasi pechokwadi mutoro uye data mamiriro.
  3. Chishandiso chekutarisa zvikanganiso, kudonha, uye mushandisi maitiro.
  4. Gadzirira nzira dzekudzosera kumashure uye dzezviitiko usati wawedzera.

Ramba Uchiongorora

Tsamba yemahara

Tora iyo yemazuva ese AI pfupiso

Nhatu dzakasimbiswa dzeAI nyaya mangwanani ega ega pakati pevhiki, dzakanyorwa neChirungu chakajeka. Yemahara zvachose, hapana ads.

Imwe email zuva rega rega revhiki. Bvisa kunyoresa kamwe chete. Hatimbotengese kana kugovera kero yako.

Zviedze iwe pachako

Take the Optimizer State Offloading to CPU and NVMe quiz

Mhinduro yepakarepo pamhinduro yega yega, uye chitupa chinogovaniswa chine ID inobatika kana wapedza kosi.

Tanga mibvunzo

Tsigira dzidzo yemahara yeAI. AI Understanding ndeye 501(c)(3) isingabatsiri - hapana ads, hapana paywall, zvachose. Ita chipo

Mibvunzo inowanzo bvunzwa

What is Optimizer State Offloading to CPU and NVMe?

Chirevo chekuchengetedza ndangariro chinomisa kuchengetwa kwebhuku kunorema kwekudzidziswa (optimizer states, gradients, dzimwe nguva uremu) muCPU RAM kana paNVMe SSDs pane kushomeka GPU memory. Inoita kuti vanhu vadzidzise mamodheru akakurisa kupfuura ndangariro dzeGPU yavo yaizobvumira.

Ndechipi chinangwa chekutanga chekuburitsa optimizer nyika kuCPU kana NVMe?

Kuburitsa kunofambisa kunorema optimizer kunobva paGPU kupinda muCPU RAM kana NVMe, kusunungura ndangariro yeGPU kuti mamodheru akakura agone kudzidziswa pane imwechete hardware.

For an Adam setup with two FP32 moments and an FP32 master weight, how much memory do those three values use per parameter?

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.

Ndeipi dhizaini inoratidza yakakurumbira yakakura-chiyero optimizer kurodha?

DeepSpeed's ZeRO-Offload uye ZeRO-Infinity yakaunzwa uye yakakurumbira kuburitsa optimizer nyika kuCPU neNVMe pachiyero.

Ndeupi mutengo mukuru wekuita kuburitsa kuCPU kana NVMe?

Kufambisa nyika kure-GPU kunoreva kuendesa data pamusoro pePCIe kana kuNVMe, inononoka pane pa-GPU ndangariro, saka madonhwe ekubuda kunze kwekunge akaputirana nekomputa.

Masisitimu ekurodha anoviga sei yakawanda yekutamisa latency?

Vagadziri vanofanotungamira zvikamu zvinodikanwa pamusoro pePCIe iyo GPU ichiri komputa, inopindirana kutaurirana nekomputa kumasiki latency.