LLM大模型实战 —— DB-GPT阿里云部署指南

农妇山泉一亩田  金牌会员 | 2024-8-4 18:16:55 | 来自手机 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 984|帖子 984|积分 2952

简介:  DB-GPT 是一个实验性的开源应用,它基于FastChat,并利用vicuna-13b作为基础模型, 模型与数据全部本地化部署, 绝对保障数据的隐私安全。 同时此GPT项目可以直接本地部署连接到私有数据库, 进行私有数据处理, 目前已支持SQL生成、SQL诊断、数据库知识问答、数据处理等一系列的工作。
背景

项目所在: https://github.com/csunny/DB-GPT
DB-GPT从5月6号正式发布第一个可运行版本,到目前刚刚经过一周的时间,引起了各人的广泛关注。 但是由于DB-GPT是用的Vicuna-13B的模型作为Base Model,以是很多同学在部署过程中遇到了较大的困难。DB-GPT在斲丧级GPU即可完成部署, 具体部署的硬件说明如下:
  1. GPU型号 |  显存大小 |   性能
  2. -------|----------|------------------------------
  3. TRX4090| 24G      |可以流畅的进行对话推理,无卡顿
  4. TRX3090| 24G      |可以流畅进行对话推理,有卡顿感,但好与V100
  5. V100   | 16G      | 可以进行对话推理,有明显卡顿
复制代码

在本文中,为了方便各人部署测试,以是我们基于阿里云写了个完整的部署教程,感兴趣的同学可以基于此教程进行部署实践,由于项目主要是实验性质,不建议生产环境直接利用。
环境准备

创建阿里云GPU实例


  • 阿里云账号准备, 如果没有阿里云账号,首先需要在阿里云注册一个账号,并充值100元。充值可以在右上角账号中心进行操作。


  • 充值好之后,我们就可以去购买GPU服务器了。找到GPU云服务器产品,点击购买,进入到选配页面,在如下页面,选择按量付费,然后选择图中所示的型号即可。如果对推理性能有要求,可以选择更高的配置。

选择型号之后,还需要选择系统、GPU驱动、SSD存储、以及实例释放策略等选项。这里需要注意,云盘的规格不要选择太小,建议不小于200G, 同时去掉随实例释放的选项,这样后续我们可以只释放实例,模型数据可以保留。如果此实例,你想长期利用,建议做个快照备份。固然,思量到代价原因,也可以挂一块NFS来存储数据。

选择完以上步调之后,点击下一步进入到网络与安全组配置界面,这里注意的是网络带宽我们可以选择大一点。这里我选择的是25M的网络。注意我们要选择按流量付费, 同时需要去买个流量包。

选择好之后,进入下一步, 系统配置界面,注意要选择一下自界说密码,需要设置一下root密码,如果这一步未设置,等机器启动之后设置也可。

这一步结束之后,分组设置无需特别设置,直接进入下一步即可。 最后在确认订单界面,我们需要重点设置一下自动释放时间,防止后续忘记释放造成连续的费用。选择好之后,直接点击创建实例,开实例阶段就结束了。

登岸实例安装环境依赖

进入到管理控制台,实例界面,点击长途连接,然后直接登岸,在登岸界面输出对应的密码,即可完成登岸。进入到管理控制台之后,首先会安装对应的GPU显卡驱动。显卡驱动安装完成之后,我们需要安装以下须要的依赖包
  1. yum -y install git
  2. yum install git-lfs
  3. yum install g++
  4. yum install docker
复制代码
如上,我们的环境准备就完成了,接下来我们开始部署服务。
部署服务

部署服务有以下几步。1. 模型准备 2. 依赖服务启动 3. DB-GPT服务启动
模型准备

在准备模型阶段,我们首先要下载模型。 这里我们依赖两个基础模型,分别是vicuna-13B以及sentence-transformer
  1. git clone https://github.com/csunny/DB-GPT.git
  2. git clone https://huggingface.co/Tribbiani/vicuna-13b
  3. https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2
复制代码
模型下载好之后,模型文件需要防止到代码路径下。

   DB-GPT/models/
  依赖服务启动

我们的项目由于要直接连接数据库,这里我们以MySQL作为样例。前面我们已经安装好了Docker,这里我们只需要通过docker命令启动MySQL即可
   docker run --name=mysql -p 3306:3306 -e MYSQL_ROOT_PASSWORD=aa12345678 -dit mysql:latest
  向量数据库我们默认利用的是Chroma内存数据库,以是无需特别安装,如果有需要连接其他的同学,可以按照我们的教程进行安装配置。
DB-GPT服务启动

首先安装环境与依赖

python 环境我们要起是大于3.9,这里我们用3.10来进行安装。
首先我们需要安装conda环境,我们利用miniconda3
  1. wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.3.1-0-Linux-x86_64.sh
  2. sh Miniconda3-py310_23.3.1-0-Linux-x86_64.sh
复制代码
如上,我们就安装conda环境完成了,安装完成之后,需要见效一下环境变量。
  1. source /root/.bashrc
复制代码
  1. conda create -n dbgpt_env python=3.10
  2. conda activate dbgpt_env
复制代码
如图所示,我们即进入到了我们的环境,接下来安装pip依赖运行就可以啦。

  1. pip install -r requirements.txt
复制代码
安装完成之后我们就可以运行了,但是运行过程中我们发现报了以下错误。 我们通过添加.pth的方式指定一下环境路径。
  1. echo "/root/workspace/DB-GPT" > /root/miniconda3/env/dbgpt_env/lib/python3.10/site-packages/dbgpt.pth
复制代码
运行命令启动服务端:
  1. python pilot/server/vicuna_server.py
复制代码
如下图所示,我们的服务就启动成功了,接下来我们来启动客户端。

运行命令启动客户端
  1. python pilot/server/webserver.py
复制代码
运行过程中,我们发现报了以下错误。 表明我们的mysql容器未创建成功,需要重新创建一下。

需要注意,mysql容器创建的密码是否与代码配置中的密码一致。 如果不一致可以重新创建大概修改代码。如图所示,我们的客户端服务就启动起来了。

配置安全策略

虽然我们服务已经启动了,但是我们还是无法通过公网访问,我们需要配置安全策略,将端口开放出来。
在阿里云管理控制台,我们找到安全组 -> 管理规则。
通过手动添加开启如下三个规则。

好了,到这里我们就可以访问我们的服务了。如果你需要在本地启动代码,长途连接到阿里云的服务器,只需要在阿里云启动一个后端服务,前端服务本地启动即可。 需要注意要修改本地服务的VICUNA_MODEL_SERVER IP为你的阿里云IP。 然后在本地只启动前端服务即可:
   python pilot/server/webserver.py

  演示

通过服务器公网ip,我们可以在欣赏器打开我们的服务。剩下的就是利用了,我们简单做个小演示吧。

原生回答


基于知识库


小结

本教程中介绍了如安在阿里云上通过申请GPU来部署DB-GPT服务

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

农妇山泉一亩田

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表