phi-2 の GPU 要件
phi-2 のパラメータ数は 2.8B です。推論で動かすには、16 ビットで約 6.7 GB、8 ビットで約 3.3 GB、4 ビットで約 1.7 GB の GPU メモリを見込んでください。KV キャッシュと活性化の分として約 20% を含みます。
ファインチューニングに必要なもの
QLoRA は 4 ビットで固定したモデルの上でアダプタを学習し、LoRA は 16 ビットで固定したモデルの上で学習します。フルファインチューニングはすべての重みを更新します。数値はアクティベーションを含まない値で、アクティベーションはバッチサイズとシーケンス長に応じて増えます。
| メソッド | メモリと最小構成 |
|---|---|
| QLoRA | 1.9 GB：RTX 4090 または RTX 3090 1 枚、RTX 3090 は GPU 1 時間あたり $0.19 から |
| LoRA | 6.7 GB：RTX 4090 または RTX 3090 1 枚、RTX 3090 は GPU 1 時間あたり $0.19 から |
| フルファインチューニング | 44 GB：L40S、RTX 6000 Ada または RTX A6000 1 枚、L40S は GPU 1 時間あたり $0.44 から |
モデルについて
Hugging Face 上のモデルカードとファイルから。このページのメモリの数字は GPU メモリガイドの計算に従っています。
- パラメータ数
- 2.8B
- コンテキスト長（トークン）
- 2,048
- 公開時の精度
- BF16
- ライセンス
- mit
- 公開日
- 2023-12-13
Arrakis でデプロイする
コンソールで Models を開き、vLLM、SGLang、llama.cpp のいずれかを選んで、収まる GPU を借ります。専用の API キー付きの OpenAI 互換エンドポイントと、マシンへの root SSH が使えます。GPU メモリが足りるマシンなら、自分で動かすこともできます：
vLLM
vllm serve microsoft/phi-2 --tensor-parallel-size 1よくある質問
すべての GPU の最新価格は GPU 価格 のページにあります。
phi-2 にはどれだけの GPU メモリが必要ですか？
推論で動かすには、16 ビットで約 7 GB、8 ビットで約 3 GB、4 ビットで約 2 GB です。KV キャッシュと活性化の分として約 20% を含みます。ファインチューニングには、QLoRA で約 2 GB、LoRA で約 7 GB、Adam によるフルファインチューニングで約 44 GB が必要です（活性化を除く）。
phi-2 は GPU 1 枚で動きますか？
はい。16 ビットで約 6.7 GB 必要で、RTX 4090 または RTX 3090 1 枚に収まります。
phi-2 を動かす費用はいくらですか？
16 ビットなら RTX 4090 または RTX 3090 1 枚、4 ビットなら RTX 4090 または RTX 3090 1 枚が必要です。このページの各構成には、Arrakis での現在の GPU 1 時間あたりの最安値を表示しています。料金は前払い残高から 1 時間単位で支払います。
Arrakis で phi-2 をデプロイするには？
Arrakis コンソールで Models を開き、microsoft/phi-2 を貼り付け、vLLM、SGLang、llama.cpp のいずれかを選んで、収まる GPU を借ります。専用の API キー付きの OpenAI 互換エンドポイントと、マシンへの root SSH が使えます。