一文解析如何使用vLLM部署llama3模型!
一、使用 vllm 部署 Llama3-8b-Instruct
- 交互式建模(PAI-DSW)资源地址:
https://github.com/vllm-project/vllm
vLLM 是一个用于大型语言模型(LLM)推理和服务的快速且易于使用的库。

vLLM 的快速特性包括:
- 先进的服务吞吐量
- 使用 PagedAttention 高效管理注意力机制的键和值内存
- 对传入请求进行持续批处理
- 使用 CUDA/HIP 图快速执行模型
- 量化技术:GPTQ、AWQ、SqueezeLLM、FP8 KV 缓存
- 优化的 CUDA 内核
二、下载llama3模型文件
/mnt/workspace路径下执行
mkdir models
/mnt/workspace/models路径下执行
pip install modelscope
git clone https://www.modelscope.cn/LLM-Research/Meta-Llama-3-8B-Instruct.git
使用 SHA-256 算法检查文件(可选)
shasum -a 256 model-00001-of-00004.safetensors
shasum -a 256 model-00002-of-00004.safetensors
shasum -a 256 model-00003-of-00004.safetensors
shasum -a 256 model-00004-of-00004.safetensors
2.1 安装vLLM
conda create -n vllm python=3.10
conda activate vllm
pip install vllm
pip install modelscope
2.2 模型推理
completion模式
- 服务部署
python -m vllm.entrypoints.openai.api_server --model /mnt/workspace/Meta-Llama-3-8B-Instruct --dtype auto --api-key 123456
- 服务测试 (vllm_completion_test.py
from openai import OpenAI
client = OpenAI(base_url="http://localhost:8000/v1",api_key="123456",)
print("服务连接成功")
completion = client.completions.create(
model="/mnt/workspace/Meta-Llama-3-8B-Instruct",
prompt="北京是",
max_tokens=128,
)
print("### 北京是: ")
print("Completion result: ", completion)
另外一个terminal窗口执行
conda activate vllm
python vllm_completion_test.py
2.3 chat模式
- 服务部署
python -m vllm.entrypoints.openai.api_server --model /mnt/workspace/Meta-Llama-3-8B-Instruct --dtype auto --api-key 123456
- 服务测试(vllm_chat_test.py)
from openai import OpenAI
client = OpenAI(
base_url="http://localhost:8000/v1",
api_key="123456",
)
print("服务连接成功")
completion = client.chat.completions.create(
model="/mnt/workspace/models/Meta-Llama-3-8B-Instruct",
messages=[
{"role":"system","content":"你是一位智能助手."},
{"role":"user","content":"中国的首都是哪里?"}
],
max_tokens =128,
)
print(completion.choices[0].message)
另外一个terminal窗口执行
python vllm_chat_test.py
如何系统学习掌握AI大模型?
AI大模型作为人工智能领域的重要技术突破,正成为推动各行各业创新和转型的关键力量。抓住AI大模型的风口,掌握AI大模型的知识和技能将变得越来越重要。
学习AI大模型是一个系统的过程,需要从基础开始,逐步深入到更高级的技术。
这里给大家精心整理了一份
全面的AI大模型学习资源,包括:AI大模型全套学习路线图(从入门到实战)、精品AI大模型学习书籍手册、视频教程、实战学习、面试题等,资料免费分享!

1. 成长路线图&学习规划
要学习一门新的技术,作为新手一定要先学习成长路线图,方向不对,努力白费。
这里,我们为新手和想要进一步提升的专业人士准备了一份详细的学习成长路线图和规划。可以说是最科学最系统的学习成长路线。

2. 大模型经典PDF书籍
书籍和学习文档资料是学习大模型过程中必不可少的,我们精选了一系列深入探讨大模型技术的书籍和学习文档,它们由领域内的顶尖专家撰写,内容全面、深入、详尽,为你学习大模型提供坚实的理论基础。(书籍含电子版PDF)

3. 大模型视频教程
对于很多自学或者没有基础的同学来说,书籍这些纯文字类的学习教材会觉得比较晦涩难以理解,因此,我们提供了丰富的大模型视频教程,以动态、形象的方式展示技术概念,帮助你更快、更轻松地掌握核心知识。

4. 大模型项目实战
学以致用 ,当你的理论知识积累到一定程度,就需要通过项目实战,在实际操作中检验和巩固你所学到的知识,同时为你找工作和职业发展打下坚实的基础。

5. 大模型面试题
面试不仅是技术的较量,更需要充分的准备。
在你已经掌握了大模型技术之后,就需要开始准备面试,我们将提供精心整理的大模型面试题库,涵盖当前面试中可能遇到的各种技术问题,让你在面试中游刃有余。

全套的AI大模型学习资源已经整理打包,有需要的小伙伴可以
微信扫描下方CSDN官方认证二维码,免费领取【保证100%免费】

更多推荐


所有评论(0)