LLMのGPUメモリ要求推定
モデル名を入力してください。改行して複数のモデル名を指定できます。
モデル名から推測されるパラメーター数と量子化ビット数に基づいてGPUのVRAMのメモリ要求とGPUを使用しない場合のメインメモリの使用量を推計します。
量子化表記は 4bit / Int8 / bf16 に加え、FP8・NVFP4・MXFP4 などのブロック型FPフォーマットや GGUF の Q4_K_M などからも推定します(NVFP4/MXFP4 は公称4bit、FP8/E4M3/E5M2 は8bitとして扱います)。
MoE(Mixture of Experts)モデルは 30B-A3B のように「総パラメータ数」と「有効パラメータ数」を併記します。全エキスパートをメモリに載せる必要があるためGPU(VRAM)・重みメモリは総パラメータ数基準、1トークンで実際に動くのは一部のため活性化メモリは有効パラメータ数基準で計算します。
実際のハードウェア要求とは異なる場合があります。
CPU推論時のメモリ使用量は中間活性化メモリを加算する必要があり、レイヤー数、シーケンス長、隠れ次元数、データタイプのサイズを元に近似式で計算できますが、このプログラムでは簡略化して有効パラメータ数×量子化ビット数の0.5~0.8倍であると仮定しています。また、バッチサイズは常に1です。