AI 智能对话 - ChatGLM2-6B 本地搭建入门_rtx3070 chatglm2-6b

前情提要

这一个月来干了啥事情呢？AI 绘画搞了2周左右，SD 建筑绘图，训练 LORA ，模型控制基本也上手了，可以按照预期生成自己想要的东西，那种控制感是挺开心的，不然你输入一句话生成 AI 图片完全凭天意，这叫无效沟通好不啦；突发奇想要搞 AI 歌曲合成，为了不引发版权诉讼问题，用自己的声音合成了些歌曲，因为不是在录音棚录音的，并且声音数据比较少，所以训练效果还是有点差强人意，不过相较我自己唱的而言已经好到天上去了；AI 语言的实践完成以后，我又开始对 AI 人工智能对话开始下手了，这个才是和大脑思考最接近的一项技术了，应用面也很广泛的，所以一定要搞下来。

那这些小玩意实践过之后，怎么玩呢，其实 AI 的话行业应用就两个层面了，一个是 AI 算法和应用，还有就是 AI 算力调度，因为去年参与过移动云的"东数西算"架构设计，AI 算力调度这块应该还好(去年算力资源真心浪费了)，算法逻辑那块勉强看下吧，况且老子下半年还有考试呢，不能在这上面浪费太多时间，最近两个月的周末都耗在这上面了，不能这么沉沦下去了。

搭建部署

环境信息

**
OS：Win11
GPU：3070-RTX 32G
PYTHON：3.10
**

安装依赖

项目下载

# 下载
git clone https://github.com/THUDM/ChatGLM2-6B

# 安装相关依赖
cd ChatGLM2-6B
pip install -r requirements.txt -i https://pypi.douban.com/simple
1
2
3
4
5
6

模型下载

云盘链接：https://pan.baidu.com/s/1AIerQMpvw7yO34Gq9BFxAQ
提取码：5uzo

将下载的模型放到 THUDM 文件夹下:

显卡驱动

# 查看本机显卡信息
nvidia-smi
1
2

安装 cuda-toolkit 工具: https://developer.nvidia.com/cuda-toolkit-archive

备注: 选择不高于上述CUDA的版本,建议版本为 11.8;

** Pytorch 依赖**

# 下载对应版本的 Pytorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
1
2

** CUDA 校验**

# cuda 可用校验(返回 True 则表示可用)
python -c "import torch; print(torch.cuda.is_available());"
1
2

配置选择

** 精度选择 **
在 api.py cli_demo.py web_demo.py web_demo.py 等脚本中,命令如下：

# 模型默认以 FP16 精度加载，运行模型需要大概 13GB 显存
model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True).cuda()

# 如果 GPU 显存有限，按需修改，目前只支持 4/8 bit 量化
model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True).quantize(4).cuda()

# 如果内存不足，可以直接加载量化后的模型
model = AutoModel.from_pretrained("THUDM/chatglm2-6b-int4",trust_remote_code=True).cuda()

# 如果没有 GPU 硬件的话，也可以在 CPU 上进行对话，但是对话速度会很慢，需要32GB内存（量化模型需要5GB内存）
model = AutoModel.from_pretrained("THUDM/chatglm2-6b", trust_remote_code=True).float()

# 如果内存不足，可以直接加载量化后的模型
model = AutoModel.from_pretrained("THUDM/chatglm2-6b-int4",trust_remote_code=True).float()

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

在 CPU 上运行量化后的模型，还需要安装 gcc 与 openmp。多数 Linux 发行版默认已安装。对于 Windows ，可在安装 TDM-GCC 时勾选 openmp。

启动运行

** 控制台 **

# 控制台运行
python cli_demo.py
1
2

** 网页版 **

# 方式一
python web_demo.py

# 方式二
# 安装 streamlit_chat 模块
pip install streamlit_chat -i https://pypi.douban.com/simple
streamlit run web_demo2.py
1
2
3
4
5
6
7

说明: 由于国内 Gradio 的网络访问较为缓慢，启用 demo.queue().launch(share=True, inbrowser=True) 时所有网络会经过 Gradio 服务器转发，导致打字机体验大幅下降，现在默认启动方式已经改为 share=False，如有需要公网访问的需求，可以重新修改为 share=True 启动。

** API **

# 安装 fastapi uvicorn 模块
pip install fastapi uvicorn -i https://pypi.douban.com/simple

# 运行 api
python api.py
1
2
3
4
5

默认部署在本地的 8000 端口，通过 POST 方法进行调用。

curl -X POST "http://127.0.0.1:8000" \
     -H 'Content-Type: application/json' \
     -d '{"prompt": "你好", "history": []}'
1
2
3

总结

附录

参考：https://juejin.cn/post/7250348861238870053