AI(人工知能)6分で読めます

GPU不要で2.8兆パラメータモデルを旧型サーバーで動作、Optaneメモリが鍵に

AI要約複数の情報源を統合して作成しています

40コアの旧型XeonサーバーとOptane Persistent Memoryを使い、GPUなしでMoonshot AIの2.8兆パラメータモデル「Kimi K3」を動かした事例が共有された。生成速度は毎秒0.85トークン。

GPU不要で2.8兆パラメータモデルを旧型サーバーで動作、Optaneメモリが鍵に

人工知能のモデル規模が拡大を続けるなか、数兆パラメータ級の大規模モデルは高価なハイエンドGPUクラスタがなければ動かせないのが一般的です。そんな常識に挑んだ事例がコミュニティで共有されました。GPUを搭載していない40コアの旧型サーバーだけで、Moonshot AI(月之暗面)の2.8兆パラメータモデル「Kimi K3」の動作に成功したというものです。

ポイント

  • 使用したのは40コアの第3世代Xeonスケーラブルプロセッサ搭載の旧型サーバーで、GPUは非搭載
  • Kimi K3は総パラメータ数2.8兆の混合エキスパート(MoE)モデル。1トークンあたり約1040億パラメータが活性化する
  • 公式スペックのコンテキスト長は104万トークン
  • 生産終了したIntelのOptane Persistent Memory 200シリーズを組み合わせ、1枚あたり最大512GBを確保
  • 1ビット量子化時の生成速度は毎秒0.85トークン、Direct Access(DAX)モードをオフにすると最大毎秒1.5トークン

ハードウェアの制約と対策

Kimi K3は総パラメータ数2.8兆の混合エキスパート(MoE)モデルです。各トークンで活性化されるのは約1040億パラメータにとどまりますが、公式スペックのコンテキスト長は104万トークンに達します。

この規模のデータを扱うには、1ビット量子化まで圧縮しても数百GBのメモリが必要です。一般家庭用マザーボードの搭載限界を超えるため、投稿者は40コアの第3世代Xeonスケーラブルプロセッサを搭載したサーバーに、NVIDIAが生産終了したOptane Persistent Memory 200シリーズを組み合わせました。1枚あたり最大512GBという大容量を確保する構成です。

ソフトウェアの調整

ソフトウェア面ではllama.cppを利用しました。Optane Persistent MemoryをApp Directモードに設定し、ハードディスクのキャッシュとしてシステムに任せるのではなく、プログラムが直接読み書きするメモリとして使っています。

ただし、この非主流の低コスト構成には速度面の代償があります。1ビット量子化での実測生成速度は毎秒0.85トークンです。Direct Access(DAX)モードをオフにすると、速度は最大でも毎秒1.5トークンにとどまります。

出典元記事を読む(AI(人工知能))
LINEX
NihonTechHub

NihonTechHub

技術系ジャーナリスト

日本の最新テクノロジーやスタートアップ情報を発信するプラットフォームです。国内外のイノベーションをつなぎ、未来を切り開くための知識とインスピレーションを提供します。

Tools

NihonTechHub ソフトウェア · 私たちが開発した便利なツールとソフトウェア

Google Drive ダウンローダー

閲覧専用ファイルの簡単ダウンロード

Google DriveのView-only(閲覧専用)ファイルを簡単にダウンロードできるツールです。PDF、Docxファイルに対応しています。

PDF 対応済みDocx 対応済み無料で利用可能

この機能はNihonTechHubアプリでもご利用いただけます(Android・iOS対応)

App

無料

外出先でも最新のテクノロジーニュースをチェック!リアルタイム通知とパーソナライズされたニュースフィードを受け取るにはアプリをダウンロードしてください。

  • ・リアルタイム更新
  • ・プッシュ通知
  • ・カスタマイズ可能