Arrakis / 模型 / Llama-2-7b-hf

Llama-2-7b-hf 的 GPU 需求

Llama-2-7b-hf 有 6.7B 參數。部署推理時,16 位約需 16 GB 視訊記憶體,8 位約 8.1 GB,4 位約 4.0 GB,其中包含約 20% 用於 KV 快取和啟用值。

部署推理需要什麼

推理執行它所需的視訊記憶體,能放下它的最小 GPU 配置,以及 Arrakis 當前的最低價格。

精度視訊記憶體與最小配置
16 位16 GB:一張 RTX 4090 或 RTX 3090,RTX 3090 每 GPU 小時 $0.19 起
8 位8.1 GB:一張 RTX 4090 或 RTX 3090,RTX 3090 每 GPU 小時 $0.19 起
4 位4.0 GB:一張 RTX 4090 或 RTX 3090,RTX 3090 每 GPU 小時 $0.19 起

微調需要什麼

QLoRA 在凍結的 4 位模型上訓練介面卡,LoRA 基於 16 位模型,全量微調則會更新所有權重。以上數字不含激活值,激活值會隨批大小和序列長度增長。

方法視訊記憶體與最小配置
QLoRA4.7 GB:一張 RTX 4090 或 RTX 3090,RTX 3090 每 GPU 小時 $0.19 起
LoRA16 GB:一張 RTX 4090 或 RTX 3090,RTX 3090 每 GPU 小時 $0.19 起
全量微調108 GB:一張 H200,每 GPU 小時 $3.44 起

關於這個模型

來自它在 Hugging Face 上的模型卡和檔案。本頁的視訊記憶體數字遵循GPU 視訊記憶體指南。

引數量
6.7B
釋出精度
BF16
許可證
llama2
釋出日期
2023-07-13
權重
meta-llama/Llama-2-7b-hf

在 Arrakis 上部署

在控制台中開啟 Models,選擇 vLLM、SGLang 或 llama.cpp,然後租用一塊能放下它的 GPU。你會得到一個相容 OpenAI 的端點,帶有專屬 API 金鑰,並可透過 root SSH 登入機器。也可以在任何視訊記憶體足夠的機器上自己執行:

vLLM
vllm serve meta-llama/Llama-2-7b-hf --tensor-parallel-size 1

常見問題

所有 GPU 的實時價格見 GPU 價格頁面。

Llama-2-7b-hf 需要多少視訊記憶體?
推理部署時,16 位約需 16 GB,8 位約 8 GB,4 位約 4 GB,其中包含約 20% 用於 KV 快取和啟用值。微調時,QLoRA 約需 5 GB,LoRA 約需 16 GB,使用 Adam 全量微調約需 108 GB,均未計入啟用值。
Llama-2-7b-hf 能在單張 GPU 上執行嗎?
可以。16 位下約需 16 GB,一張 RTX 4090 或 RTX 3090 就能放下。
執行 Llama-2-7b-hf 要多少錢?
16 位下需要一張 RTX 4090 或 RTX 3090;4 位下需要一張 RTX 4090 或 RTX 3090。本頁每種配置都顯示 Arrakis 目前每 GPU 小時的最低價格,從預付餘額中按小時扣費。
如何在 Arrakis 上部署 Llama-2-7b-hf?
在 Arrakis 控制台中開啟 Models,貼上 meta-llama/Llama-2-7b-hf,選擇 vLLM、SGLang 或 llama.cpp,然後租用一張能放下它的 GPU。你會得到一個相容 OpenAI 的端點,帶有專屬 API 金鑰,並可透過 root SSH 登入機器。