大模型部署手记(2)baichuan2+Windows GPU

1.简介

组织机构:百川智能(前搜狗CEO王小川创立)

代码仓:GitHub - baichuan-inc/Baichuan2: A series of large language models developed by Baichuan Intelligent Technology

模型:baichuan-inc/Baichuan2-7B-Chat-4bits

下载:https://huggingface.co/baichuan-inc/Baichuan2-7B-Chat-4bits

镜像下载:https://aliendao.cn/models/baichuan-inc/Baichuan2-7B-Chat-4bits

硬件环境:暗影精灵7Plus

Windows版本:Windows 11家庭中文版 Insider Preview 22H2

内存 32G

GPU显卡:Nvidia GTX 3080 Laptop (16G)

参考资料:https://zhuanlan.zhihu.com/p/658999248

2.代码和模型下载

下载代码仓:

git clone GitHub - baichuan-inc/Baichuan2: A series of large language models developed by Baichuan Intelligent Technology

大模型部署手记(2)baichuan2+Windows GPU_第1张图片

打开 https://huggingface.co/baichuan-inc/Baichuan2-7B-Chat-4bits 下载模型

并将其拷贝到 d:\Baichuan2\baichuan-inc\Baichuan2-7B-Chat-4bits 目录

大模型部署手记(2)baichuan2+Windows GPU_第2张图片

3.安装依赖

创建conda环境

conda create -n baichuan2 python=3.10

conda activate baichuan2

大模型部署手记(2)baichuan2+Windows GPU_第3张图片

cd Baichuan2

安装量化包:

pip install bitsandbytes --prefer-binary --extra-index-url=https://jllllll.github.io/bitsandbytes-windows-webui

大模型部署手记(2)baichuan2+Windows GPU_第4张图片

大模型部署手记(2)baichuan2+Windows GPU_第5张图片

安装Pytorch 2.0.1 for CUDA

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

大模型部署手记(2)baichuan2+Windows GPU_第6张图片

安装加速包:

pip install xformers

大模型部署手记(2)baichuan2+Windows GPU_第7张图片

将代码 cli_demo.py 改成4bit量化的模型:

大模型部署手记(2)baichuan2+Windows GPU_第8张图片

d:

cd d:\Baichuan2

pip install -r requirements.txt

大模型部署手记(2)baichuan2+Windows GPU_第9张图片

4.部署验证

运行命令行模式:

python cli_demo.py

做一些简单的交互:

大模型部署手记(2)baichuan2+Windows GPU_第10张图片

修改web_demo.py文件:

大模型部署手记(2)baichuan2+Windows GPU_第11张图片

运行网页模式:

python web_demo.py

大模型部署手记(2)baichuan2+Windows GPU_第12张图片

这里好像哪里不对,但是系统提示可以使用streamlit运行:

streamlit run web_demo.py

系统自动打开浏览器:

大模型部署手记(2)baichuan2+Windows GPU_第13张图片

大模型部署手记(2)baichuan2+Windows GPU_第14张图片

做一些简单的交互:

大模型部署手记(2)baichuan2+Windows GPU_第15张图片

大模型部署手记(2)baichuan2+Windows GPU_第16张图片

(全文完,谢谢阅读)

你可能感兴趣的:(大模型,深度学习)