手把手教你如何部署ChatGLM-6B大模型

最近咸鱼时间学习了大模型的知识,正好不上班,趁着这时间写一下自己的实践供大家参考 

云服务器环境搭建

目前阿里云服务器有免费的使用机会,大家可以申请用来学习和实验,账号的话:支付宝啥的都可以直接注册。

具体操作步骤:进入阿里云服务器网站,找到“产品”,然后找到“机器学习平台PAI”,点击后就会下面的图,这边点击免费试用即可。

手把手教你如何部署ChatGLM-6B大模型_第1张图片

 服务器的环境我是根据b站up主学习的,这里我们选择第一个交互式建模,点击立即试用

手把手教你如何部署ChatGLM-6B大模型_第2张图片

然后我们点击交互式建模,点击“创建实例”,

手把手教你如何部署ChatGLM-6B大模型_第3张图片

 实例的地域、工作空间、名称这些任意,重点是GPU和环境,GPU可以选择V100的卡,应该是够用了,然后镜像选择,都在下面图列出来了,按照这个走就行。数据集和网络配置先不要选择!!!

手把手教你如何部署ChatGLM-6B大模型_第4张图片

手把手教你如何部署ChatGLM-6B大模型_第5张图片

模型的下载

这边需要下载两个东西,分别是chatGLM-6B的github地址和模型地址,分别列出来了:

chatGLM-6B的Github地址:https://github.com/THUDM/ChatGLM-6B

chatGLM-6B的模型地址:https://huggingface.co/THUDM/chatglm-6b

模型下载我这边用的两个方法:一个是下载到本地,然后上传到服务器上;另一个是直接git clone的方式。

下载到本地:

大家都懂得,直接在github的界面,下载压缩包就行,然后放到服务器中解压就会有chatGLM-6B文件夹。但是模型的话,需要把相关文件都要下载下来,部分文件比较小,下载也比较慢!如图所示,把所有文件下载然后上传到服务器就行

手把手教你如何部署ChatGLM-6B大模型_第6张图片

Git:

在这之前,有些操作需要执行,我列出来了

更新git-lfs:

apt-get update

apt-get install git-lfs

初始化git:

git init 

git lfs install

然后我们就可以下载chatGLM-6B源码了

git clone https://github.com/THUDM/ChatGLM-6B.git

进入到chatGLM-6B文件夹,可以看到有一个requirements文件夹,所以我们需要搭建相关的环境

手把手教你如何部署ChatGLM-6B大模型_第7张图片

 直接pip install -r requirements就行

创建mymodel文件夹,这里是存放模型和相关文件的,后面demo测试的模型路径是要指向这个文件夹的

mkdir model

cd model

然后进入文件夹后,就是下载模型文件了,我用git的方法一直都不行,容易出问题,推荐大家自己本地下好,然后传到这个文件夹里面去就可以

模型部署测试

当我们把相关文件都传到服务器,找到chatGLM-6B文件夹中web_demo.py文件,直接点击该文件,把路径改为服务器中你存储的模型的路径,要改的路径就是下图中我用蓝色框框圈出来的,我这里是model/chatglm-6b,记得保存哦!

手把手教你如何部署ChatGLM-6B大模型_第8张图片

然后服务器终端就可以运行模型啦,运行命令:python web_demo.py后,命令行中会出现一个网址,直接点击就会跳转到chatGLM-6B的界面了,然后就可以对话了,话不多说,看图!

手把手教你如何部署ChatGLM-6B大模型_第9张图片

手把手教你如何部署ChatGLM-6B大模型_第10张图片

 到这里,就完成了这个模型的部署运行了。我是个新人,这也是自己第一个的blog,如果有什么写的不好的地方,大家多多见谅哈,有问题欢迎随时和我交流!!!

你可能感兴趣的:(语言模型)