任务1 部署ChatGLM3

编程入门 行业动态 更新时间:2024-10-27 03:30:00

任务1 部署ChatGLM3

任务1 部署ChatGLM3

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

  • 0 介绍:
  • 1 趋动云项目创建与环境配置
    • 1.1 创建项目:
    • 1.2 配置环境
      • 1.2.1 进入终端
      • 1.2.2 设置镜像源
      • 1.2.3 克隆项目,并安装依赖
  • 2 修改代码,改路径以及启动代码
  • 3 运行代码
    • 3.1 运行gradio界面:
    • 3.2 运行streamlit界面:

0 介绍:

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

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

1.1 创建项目:

创建好账号之后,进入自己的空间,点击右上角的创建项目

给自己项目命名,选择本地代码:

镜像选择pytorch2.0.1,python3.9

选择预训练模型,点击公开,选择不要葱姜蒜上传的这个ChtaGLM3-6B模型。

都选完之后,点击右下角的创建,代码选择暂不上传。待会直接clone代码。
点击运行代码

资源配置选择:B1.large,24G的显存足够加载模型了。其他的不需要设置,然
后点击右下角的开始运行。

1.2 配置环境

1.2.1 进入终端

等右边两个工具全部加载完毕之后,再点击JupyterLabi进入开发环境

进入界面之后会有一个环境界面,可以简单运行一下环境界面:可以看到每个部分的文件都有自己固定的位置。

然后点击 小加号,新建一个终端。

点击 terminal 进入终端:

1.2.2 设置镜像源

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

tmux

升级apt,安装unzip:

apt-get update &apt-get install unzip

设置镜像源,升级pip:

git config --global url."/".insteadof https:/
pip config set global.index-url 
python3 -m pip install --upgrade pip

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

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

1.2.3 克隆项目,并安装依赖

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

git clone .git
cd ChatGLM3


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

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

pip install -r requirements.txt

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

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

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

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

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

3 运行代码

3.1 运行gradio界面:

python web_demo.py


加载完毕之后,复制外部访问的连接,到浏览器打开:

direct.virtaicloud:43779

使用谷歌打开可能会存在一些问题,可以换成ie浏览器:

这里的A,B这些错乱的字符,是我鼠标滚轮上下移动导致的,可以忽略,不影响的。

3.2 运行streamlit界面:

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

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

streamlit run web_demo2.py

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

添加端口:

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

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

更多推荐

任务1 部署ChatGLM3

本文发布于:2023-11-16 16:43:37,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1627529.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!