SeterusnyaPanduan seterusnya
Storan untuk Kluster Latihan AI
Teknikal
PANDUAN Teknikal
Slurm ialah pengurus beban kerja sumber terbuka yang menjadualkan dan menjalankan kerja pada kelompok pengkomputeran berprestasi tinggi, dan ia telah menjadi pilihan lalai untuk latihan AI yang besar.
Dengar panduan iniAI-diceritakan · ~2 min
It matters because it reliably distributes massive training runs across thousands of GPUs.
Slurm (Utiliti Linux Mudah untuk Pengurusan Sumber) berasal dari superkomputer dan kini menguasai banyak kluster latihan AI terbesar di dunia. Pengguna menyerahkan skrip kelompok dengan sbatch, meminta sumber seperti nod dan GPU dengan arahan seperti --gres=gpu:8 dan baris gilir Slurm, mengutamakan dan melancarkan kerja. Pelancar srunnya menghasilkan proses yang diselaraskan merentas nod, yang berpasangan secara semula jadi dengan rangka kerja teragih seperti PyTorch DDP dan NCCL. Slurm menjejaki perakaunan sumber, menguatkuasakan had perkongsian saksama dan pembahagian, dan mengendalikan penjadualan isian untuk memasukkan kerja kecil ke dalam jurang. Untuk latihan model sempadan, pasukan bergantung pada Slurm untuk mengurus beribu-ribu GPU, mulakan semula dari pusat pemeriksaan selepas kegagalan nod dan menyimpan kapasiti khusus untuk larian berbilang minggu yang panjang.
Makmal sempadan melancarkan latihan berbilang minggu yang dijalankan merentasi ribuan GPU dengan satu skrip sbatch yang meminta ratusan nod.
Seorang penyelidik menyerahkan 'srun --gres=gpu:8' untuk meraih lapan GPU pada satu nod untuk percubaan DDP PyTorch.
Penjadualan isian memasukkan tugas penilaian singkat ke dalam GPU terbiar sementara latihan besar yang dikhaskan menunggu untuk bermula.
Selepas nod gagal pada pertengahan larian, Slurm membuat giliran kerja dan ia menyambung semula dari pusat pemeriksaan terkini dan bukannya memulakan semula.
Slurm terus menambah awan pecah, sokongan kontena melalui Pyxis dan Enroot, dan ciri-ciri GPU yang lebih ketat. Apabila kluster AI berskala ke arah 100,000 lebih GPU, jangkakan toleransi kesalahan yang lebih kukuh, penyepaduan semula pusat pemeriksaan automatik dan kerja anjal yang mengubah saiz selepas kegagalan. Banyak organisasi kini menjalankan Slurm bersama atau di bawah Kubernetes, dan penjadual hibrid menyasarkan untuk menggabungkan kecekapan gaya HPC dengan fleksibiliti asli awan untuk larian latihan 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
Slurm ialah pengurus beban kerja sumber terbuka yang menjadualkan dan menjalankan kerja pada kelompok pengkomputeran berprestasi tinggi, dan ia telah menjadi pilihan lalai untuk latihan AI yang besar. Ia penting kerana ia boleh mengagihkan latihan besar-besaran yang dijalankan merentasi beribu-ribu GPU.
sbatch menyerahkan skrip kelompok yang menerangkan sumber dan arahan untuk sesuatu kerja ke baris gilir Slurm.
Sistem Sumber Generik (GRES) membenarkan kerja meminta GPU secara eksplisit, contohnya --gres=gpu:8.
slurmctld ialah daemon pengawal yang menjadualkan kerja, manakala slurmd berjalan pada setiap nod untuk melancarkan tugas.
srun launches task steps in the allocation and exposes Slurm task information; distributed frameworks may still require explicit rendezvous settings.
Isian semula meningkatkan penggunaan dengan memasukkan kerja yang lebih kecil ke dalam jurang penjadualan selagi ia tidak menolak kerja yang dikhaskan.
Teruskan belajar
Lebih banyak panduan dipilih untuk topik ini
SeterusnyaPanduan seterusnya
Storan untuk Kluster Latihan AI
Teknikal