任务1 部署ChatGLM3-6B大模型并进行对话测试

0 介绍:

本项目是根据datawhale 与 趋动云平台联合 联合的项目,本文主要在于学习和了解使用大模型。
使用的大模型是 ChatGLM3-6B, 这是智谱AI和清华大学KEG实验室联合发布的新一代对话预训
练模型。
整体平台感觉还是非常好用的,方便模型的部署。
项目地址: https://github.com/THUDM/ChatGLM3
image.png

1 趋动云项目创建与环境配置

1.1 创建项目:

创建好账号之后,进入自己的空间,点击右上角的创建项目
image.png
给自己项目命名,选择本地代码:
image.png
镜像选择pytorch2.0.1,python3.9
image.png
选择预训练模型,点击公开,选择不要葱姜蒜上传的这个ChtaGLM3-6B模型。
image.png

都选完之后,点击右下角的创建,代码选择暂不上传。待会直接clone代码。
点击运行代码
image.png
资源配置选择:B1.large,24G的显存足够加载模型了。其他的不需要设置,然
后点击右下角的开始运行。
image.png

1.2 配置环境

1.2.1 进入终端

等右边两个工具全部加载完毕之后,再点击JupyterLabi进入开发环境
image.png
进入界面之后会有一个环境界面,可以简单运行一下环境界面:可以看到每个部分的文件都有自己固定的位置。
image.png

然后点击 小加号,新建一个终端。
image.png
点击 terminal 进入终端:
image.png

image.png

1.2.2 设置镜像源

首先在终端输入tmux,进入一个新的会话窗口。使用tux可以保持终端的稳定性。

tmux

升级apt,安装unzip:

apt-get update &apt-get install unzip

image.png

设置镜像源,升级pip:

git config --global url."https://gitclone.com/".insteadof https:/
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple
python3 -m pip install --upgrade pip

注:如果存在一些连接错误,你可以在23行中间添加

pip config set global.trusted-host mirrors.ustc.edu.cn

image.png

1.2.3 克隆项目,并安装依赖

克隆项目:并进入项目目录:

git clone https://github.com/THUDM/ChatGLM3.git
cd ChatGLM3

image.png
我觉得应该在里面搞一个虚拟环境,然后在虚拟环境里面安装,而不是在base里面安装。

返回终端,安装依赖:
修改requirements:
双击左侧的requirements.txt文件,把其中的torch删掉,因为我们的环境中已经有
torch了,避免重复下载浪费时间。
image.png

pip install -r requirements.txt

image.png

2 修改代码,改路径以及启动代码

修改web_demo2.py ** 将加载模型的路径**修改为: …/…/pretrain 如下图所示:
image.png

修改web_demo.py文件,先修改路径代码,然后修改启动代码。
image.png

将下方的启动代码,修改为 下方代码:
demo.queue().launch(share=False,server_name=“0.0.0.0”,server_port=7000)
image.png

于此同时,在界面的右边,添加外部端口:7000
image.png

3 运行代码

3.1 运行gradio界面:

python web_demo.py

image.png
加载完毕之后,复制外部访问的连接,到浏览器打开:
image.png
direct.virtaicloud.com:43779

使用谷歌打开可能会存在一些问题,可以换成ie浏览器:
image.png
这里的A,B这些错乱的字符,是我鼠标滚轮上下移动导致的,可以忽略,不影响的。
image.png

3.2 运行streamlit界面:

如果你已经运行了gradio, 需要先杀掉这个进程,否则内存不够用。
使用 **ctrl + c **杀掉进行。
杀掉进程之后,显存不会立刻释放,可以观察右边的GPU内存占用, 查看显存释放情况。
image.png

上面已经直接修改过 web_demo2.py ,所以可以直接使用streamlit 运行:

streamlit run web_demo2.py

运行 streamlit 之后,终端会打印两个地址。在右边添加一个和终端上显示的一样的端口号
image.png
image.png

添加端口:
image.png
image.png

等待加载完毕之后,再复制到浏览器里面打开:
image.png

然后打开浏览器: 可以进行对话。
image.png

猜你喜欢

转载自blog.csdn.net/Alexa_/article/details/134223168
今日推荐